format

intel · Feb 6, 2025 · 759a84e · 759a84e
1 parent 919b65e
commit 759a84e
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/ATen/native/sparse/xpu/sycl/SparseSoftmaxKernels.cpp b/src/ATen/native/sparse/xpu/sycl/SparseSoftmaxKernels.cpp
@@ -123,11 +123,7 @@ OutputIt max_row(
 
 // Number of threads in a block given an input size up to MAX_BLOCK_SIZE
 static int getNumThreads(int nElem) {
-#if defined(USE_ROCM)
-  int threadSizes[5] = {16, 32, 64, 128, 256};
-#else
   int threadSizes[5] = {32, 64, 128, 256, 512};
-#endif
   for (int i = 0; i != 5; ++i) {
     if (nElem <= threadSizes[i]) {
       return threadSizes[i];