File tree 1 file changed +16
-0
lines changed 1 file changed +16
-0
lines changed Original file line number Diff line number Diff line change 117
117
" s_constants " : {
118
118
" RoiPooling2d_backward_kernel<<< " : " RoiPooling2d_backward_kernel<float><<<"
119
119
}
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
+ }
120
136
}
121
137
],
122
138
" disabled_modules " : [
You can’t perform that action at this time.
0 commit comments