update stream to debug

feelpp · Sep 8, 2024 · c8a5c6a · c8a5c6a
1 parent 3fad770
commit c8a5c6a
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 2 deletions.
diff --git a/Src/Cuda/SpCudaUtils.hpp b/Src/Cuda/SpCudaUtils.hpp
@@ -5,6 +5,7 @@
 #include <cstring>
 #include <vector>
 #include <sstream>
+#include <cassert>
 
 #include "Config/SpConfig.hpp"
 
@@ -64,6 +65,12 @@ class SpCudaUtils{
         return free_byte;
     }
 
+    static int getUsingDevice() {
+        int currentDevice;
+        CUDA_ASSERT(cudaGetDevice(&currentDevice));
+        return currentDevice;
+    }
+
     static void UseDevice(const int deviceId){
         if(deviceId >= GetNbDevices()){
             std::cerr << "[SPECX] Ask to use gpu " << deviceId
@@ -77,6 +84,8 @@ class SpCudaUtils{
     }
 
     static void SynchronizeStream(cudaStream_t& stream){
+        assert(stream != nullptr);
+        assert(getUsingDevice() != -1);
         CUDA_ASSERT(cudaStreamSynchronize(stream));
     }
 

diff --git a/Src/Cuda/SpCudaWorkerData.hpp b/Src/Cuda/SpCudaWorkerData.hpp
@@ -14,7 +14,7 @@
 
 struct SpCudaWorkerData {
     int cudaId = -1;
-    cudaStream_t stream;
+    cudaStream_t stream = nullptr;
 
     void init(int deviceId){
         cudaId = deviceId;

diff --git a/Src/Hip/SpHipUtils.hpp b/Src/Hip/SpHipUtils.hpp
@@ -5,6 +5,7 @@
 #include <cstring>
 #include <vector>
 #include <sstream>
+#include <cassert>
 
 #include "Config/SpConfig.hpp"
 
@@ -64,6 +65,12 @@ class SpHipUtils{
         return free_byte;
     }
 
+    static int getUsingDevice() {
+        int currentDevice;
+        HIP_ASSERT(hipGetDevice(&currentDevice));
+        return currentDevice;
+    }
+
     static void UseDevice(const int deviceId){
         if(deviceId >= GetNbDevices()){
             std::cerr << "[SPECX] Ask to use gpu " << deviceId
@@ -77,6 +84,8 @@ class SpHipUtils{
     }
 
     static void SynchronizeStream(hipStream_t& stream){
+        assert(stream != nullptr);
+        assert(getUsingDevice() != -1);
         HIP_ASSERT(hipStreamSynchronize(stream));
     }
 

diff --git a/Src/Hip/SpHipWorkerData.hpp b/Src/Hip/SpHipWorkerData.hpp
@@ -14,7 +14,7 @@
 
 struct SpHipWorkerData {
     int hipId = -1;
-    hipStream_t stream;
+    hipStream_t stream = nullptr;
 
     void init(int deviceId){
         hipId = deviceId;