Skip to content

GitLab

Unverified Commit a63f98ad authored by Jeff Daily's avatar Jeff Daily Committed by GitHub
Browse files

[ROCm] remove HCC references (#8070)

parent 3fb88b3e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/csrc/ops/cuda/deform_conv2d_kernel.cu
View file @ a63f98ad
...@@ -82,7 +82,7 @@ namespace { ...@@ -82,7 +82,7 @@ namespace {
const int kMaxParallelImgs = 32; const int kMaxParallelImgs = 32;
inline unsigned int GET_THREADS() { inline unsigned int GET_THREADS() {
#ifdef __HIP_PLATFORM_HCC__ #ifdef WITH_HIP
return 256; return 256;
#endif #endif
return 512; return 512;
......
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