fix

Bob-Chen222 · Jun 3, 2024 · d055adf · d055adf
1 parent d90b093
commit d055adf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 55 deletions.
diff --git a/lib/kernels/src/conv_2d_kernels.h b/lib/kernels/src/conv_2d_kernels.h
diff --git a/lib/kernels/src/hip/loss_function_kernels.cpp b/lib/kernels/src/hip/loss_function_kernels.cpp
@@ -77,8 +77,8 @@ void sparse_categorical_crossentropy_loss_backward_kernel(
 
   // launch kernel in hip
   hipLaunchKernelGGL(sparse_categorical_crossentropy_loss_backward,
-                     GET_BLOCKS(num_samples * k),
-                     dCUDA_NUM_THREADS,
+                     GET_BLOCKS(num_samples),
+                     CUDA_NUM_THREADS,
                      0,
                      stream,
                      logit_grad_ptr,