We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1cb818b commit 9030e36Copy full SHA for 9030e36
aten/src/ATen/native/cuda/Blas.cpp
@@ -257,9 +257,9 @@ static bool isSupportedHipLtROCmArch(int index) {
257
hipDeviceProp_t* prop = at::cuda::getDeviceProperties(index);
258
std::string device_arch = prop->gcnArchName;
259
static const std::vector<std::string> archs = {
260
- "gfx90a", "gfx942",
+ "gfx90a", "gfx942"
261
#if ROCM_VERSION >= 60300
262
- "gfx1100", "gfx1101", "gfx1200", "gfx1201"
+ , "gfx1200", "gfx1201"
263
#endif
264
#if ROCM_VERSION >= 60500
265
"gfx950"
0 commit comments