fix: fix for PRs

Signed-off-by: YdrMaster <ydrml@hotmail.com>
InfiniTensor · Jan 31, 2024 · 9169f5f · 9169f5f
1 parent 237c494
commit 9169f5f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/02hardware/src/devices/mlu/device.cc b/src/02hardware/src/devices/mlu/device.cc
@@ -25,7 +25,9 @@ namespace refactor::hardware {
     Mlu::Mlu(int32_t card) : Device(card, bangMemory(card)) {}
 
     void Mlu::setContext() const noexcept {
+#ifdef USE_BANG
         setDevice(_card);
+#endif
     }
 
 }// namespace refactor::hardware
diff --git a/src/04kernel/cuda/src/pad.cu b/src/04kernel/cuda/src/pad.cu
@@ -46,7 +46,6 @@ namespace refactor::kernel::cuda {
         unsigned int rank,
         unsigned int blockSize) {
 
-
         padKernel<<<
             params.gridSize,
             params.blockSize,