summaryrefslogtreecommitdiff
path: root/src/stream_manager.cc
diff options
context:
space:
mode:
authorJRPan <[email protected]>2025-08-18 23:09:47 -0700
committerGitHub <[email protected]>2025-08-19 06:09:47 +0000
commit21b18be3832b80b00d4890f041818bac431866e7 (patch)
tree4a8801a93a6dde20b747ef6db509215894c9f3ec /src/stream_manager.cc
parent33644740f15e0fa9f3d64ff98956569b2acf47bc (diff)
running formatter (#126)
* running formatter * consolidate CI runs * use cluster to run formatter * use cluster to run formatter * Add a CI-Success step
Diffstat (limited to 'src/stream_manager.cc')
-rw-r--r--src/stream_manager.cc39
1 files changed, 26 insertions, 13 deletions
diff --git a/src/stream_manager.cc b/src/stream_manager.cc
index d43964a..ab5b74f 100644
--- a/src/stream_manager.cc
+++ b/src/stream_manager.cc
@@ -34,21 +34,30 @@
unsigned CUstream_st::sm_next_stream_uid = 0;
-// SST memcpy callbacks, called after a stream operation is done via record_next_done()
-extern void SST_callback_memcpy_H2D_done(uint64_t dst, uint64_t src, size_t count, cudaStream_t stream);
-extern void SST_callback_memcpy_D2H_done(uint64_t dst, uint64_t src, size_t count, cudaStream_t stream);
+// SST memcpy callbacks, called after a stream operation is done via
+// record_next_done()
+extern void SST_callback_memcpy_H2D_done(uint64_t dst, uint64_t src,
+ size_t count, cudaStream_t stream);
+extern void SST_callback_memcpy_D2H_done(uint64_t dst, uint64_t src,
+ size_t count, cudaStream_t stream);
extern void SST_callback_memcpy_to_symbol_done();
extern void SST_callback_memcpy_from_symbol_done();
extern void SST_callback_cudaEventSynchronize_done(cudaEvent_t event);
extern void SST_callback_kernel_done(cudaStream_t stream);
-__attribute__((weak)) void SST_callback_memcpy_H2D_done(uint64_t dst, uint64_t src, size_t count, cudaStream_t stream) {}
-__attribute__((weak)) void SST_callback_memcpy_D2H_done(uint64_t dst, uint64_t src, size_t count, cudaStream_t stream) {}
+__attribute__((weak)) void SST_callback_memcpy_H2D_done(uint64_t dst,
+ uint64_t src,
+ size_t count,
+ cudaStream_t stream) {}
+__attribute__((weak)) void SST_callback_memcpy_D2H_done(uint64_t dst,
+ uint64_t src,
+ size_t count,
+ cudaStream_t stream) {}
__attribute__((weak)) void SST_callback_memcpy_to_symbol_done() {}
__attribute__((weak)) void SST_callback_memcpy_from_symbol_done() {}
-__attribute__((weak)) void SST_callback_cudaEventSynchronize_done(cudaEvent_t event);
+__attribute__((weak)) void SST_callback_cudaEventSynchronize_done(
+ cudaEvent_t event);
__attribute__((weak)) void SST_callback_kernel_done(cudaStream_t stream);
-
CUstream_st::CUstream_st() {
m_pending = false;
m_uid = sm_next_stream_uid++;
@@ -79,9 +88,7 @@ void CUstream_st::synchronize() {
} while (!done);
}
-bool CUstream_st::synchronize_check() {
- return m_operations.empty();
-}
+bool CUstream_st::synchronize_check() { return m_operations.empty(); }
void CUstream_st::push(const stream_operation &op) {
// called by host thread
@@ -142,14 +149,19 @@ bool stream_operation::do_operation(gpgpu_sim *gpu) {
gpu->memcpy_to_gpu(m_device_address_dst, m_host_address_src, m_cnt);
m_stream->record_next_done();
if (gpu->is_SST_mode()) {
- SST_callback_memcpy_H2D_done((uint64_t) m_device_address_dst, (uint64_t) m_host_address_src, m_cnt, m_stream->is_stream_zero_stream() ? 0 : m_stream);
+ SST_callback_memcpy_H2D_done(
+ (uint64_t)m_device_address_dst, (uint64_t)m_host_address_src, m_cnt,
+ m_stream->is_stream_zero_stream() ? 0 : m_stream);
}
break;
case stream_memcpy_device_to_host:
if (g_debug_execution >= 3) printf("memcpy device-to-host\n");
gpu->memcpy_from_gpu(m_host_address_dst, m_device_address_src, m_cnt);
m_stream->record_next_done();
- if (gpu->is_SST_mode()) SST_callback_memcpy_D2H_done((uint64_t) m_host_address_dst, (uint64_t) m_device_address_src, m_cnt, m_stream->is_stream_zero_stream() ? 0 : m_stream);
+ if (gpu->is_SST_mode())
+ SST_callback_memcpy_D2H_done(
+ (uint64_t)m_host_address_dst, (uint64_t)m_device_address_src, m_cnt,
+ m_stream->is_stream_zero_stream() ? 0 : m_stream);
break;
case stream_memcpy_device_to_device:
if (g_debug_execution >= 3) printf("memcpy device-to-device\n");
@@ -419,7 +431,8 @@ void stream_manager::add_stream(struct CUstream_st *stream) {
void stream_manager::destroy_stream(CUstream_st *stream) {
// called by host thread
pthread_mutex_lock(&m_lock);
- while (!stream->empty());
+ while (!stream->empty())
+ ;
std::list<CUstream_st *>::iterator s;
for (s = m_streams.begin(); s != m_streams.end(); s++) {
if (*s == stream) {