Skip to content

GitLab

Unverified Commit 4f51a937 authored by xiabo123's avatar xiabo123 Committed by GitHub
Browse files

Fix the support for ROCm (#2827)

parent 99a8d057
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/ops/csrc/pytorch/cuda/filtered_lrelu.cu
View file @ 4f51a937
...@@ -1678,7 +1678,7 @@ std::tuple<torch::Tensor, torch::Tensor, int> filtered_lrelu_op( ...@@ -1678,7 +1678,7 @@ std::tuple<torch::Tensor, torch::Tensor, int> filtered_lrelu_op(
int maxSharedBytes = 0; int maxSharedBytes = 0;
#ifdef MMCV_WITH_HIP #ifdef MMCV_WITH_HIP
cudaDeviceGetAttribute(&maxSharedBytes, cudaDeviceGetAttribute(&maxSharedBytes,
hipDeviceAttributeSharedMemPerBlockOptin, hipDeviceAttributeMaxSharedMemoryPerBlock,
x.device().index()); x.device().index());
#else #else
AT_CUDA_CHECK(cudaDeviceGetAttribute(&maxSharedBytes, AT_CUDA_CHECK(cudaDeviceGetAttribute(&maxSharedBytes,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment