Skip to content

Commit 3fadf87

Browse files
authored
Merge pull request #37 from jithunnair-amd/getNumGPUs_hack_for_ROCm
Hardcode getNumGPUs() to 1 for ROCm builds …
2 parents 09fd104 + 3f58e08 commit 3fadf87

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tools/amd_build/disabled_features.yaml

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,22 @@
117117
"s_constants": {
118118
"RoiPooling2d_backward_kernel<<<": "RoiPooling2d_backward_kernel<float><<<"
119119
}
120+
},
121+
{
122+
"path": "aten/src/ATen/CUDAStream.cpp",
123+
"s_constants": {
124+
# FIXME: ROCm currently does not support multi-GPU setup; and getNumGPUs runs into a seg fault
125+
# ROCm Pytorch issue: https://github.com/ROCmSoftwarePlatform/pytorch/issues/31
126+
"getCUDAHooks().getNumGPUs()": "1",
127+
}
128+
},
129+
{
130+
"path": "aten/src/ATen/Context.h",
131+
"s_constants": {
132+
# FIXME: ROCm currently does not support multi-GPU setup; and getNumGPUs runs into a seg fault
133+
# ROCm Pytorch issue: https://github.com/ROCmSoftwarePlatform/pytorch/issues/31
134+
"detail::getCUDAHooks().getNumGPUs()": "1",
135+
}
120136
}
121137
],
122138
"disabled_modules": [

0 commit comments

Comments
 (0)