Fix the support for ROCm (#2827)

4f51a937 · xiabo123 · GitHub · 99a8d057 · 4f51a937
Unverified Commit 4f51a937 authored Jun 21, 2023 by xiabo123 Committed by GitHub Jun 21, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmcv/ops/csrc/pytorch/cuda/filtered_lrelu.cu mmcv/ops/csrc/pytorch/cuda/filtered_lrelu.cu +1 -1

No files found.
--- a/mmcv/ops/csrc/pytorch/cuda/filtered_lrelu.cu
+++ b/mmcv/ops/csrc/pytorch/cuda/filtered_lrelu.cu
@@ -1678,7 +1678,7 @@ std::tuple<torch::Tensor, torch::Tensor, int> filtered_lrelu_op(
  int maxSharedBytes = 0;
 #ifdef MMCV_WITH_HIP
  cudaDeviceGetAttribute(&maxSharedBytes,
-                         hipDeviceAttributeSharedMemPerBlockOptin,
+                         hipDeviceAttributeMaxSharedMemoryPerBlock,
                         x.device().index());
 #else
  AT_CUDA_CHECK(cudaDeviceGetAttribute(&maxSharedBytes,