We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 76b963d commit 1063c0cCopy full SHA for 1063c0c
fbgemm_gpu/src/sparse_ops/sparse_group_index.cu
@@ -13,12 +13,7 @@ using Tensor = at::Tensor;
13
namespace fbgemm_gpu {
14
namespace {
15
16
-#ifdef USE_ROCM
17
-constexpr int kGroupIndexWarpSize = kWarpSize;
18
-#else
19
constexpr int kGroupIndexWarpSize = kWarpSize;
20
-#endif
21
-
22
constexpr int GROUP_INDEX_SELECT_UNROLL_FACTOR = 1;
23
constexpr int GROUP_INDEX_SELECT_COLS_PER_WARP =
24
GROUP_INDEX_SELECT_UNROLL_FACTOR * kGroupIndexWarpSize;
0 commit comments