Skip to content

Commit 1fee196

Browse files
authored
ROCm: Add trailing comma for consistency in gfx architecture (#2172)
Ported from upstream https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/150250.patch and fixed patch error Signed-off-by: Jagadish Krishnamoorthy <[email protected]>
1 parent 03c7da0 commit 1fee196

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

aten/src/ATen/Context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -362,7 +362,7 @@ at::BlasBackend Context::blasPreferredBackend() {
362362
static const std::vector<std::string> archs = {
363363
"gfx90a", "gfx942",
364364
#if ROCM_VERSION >= 60300
365-
"gfx1100", "gfx1101", "gfx1200", "gfx1201"
365+
"gfx1100", "gfx1101", "gfx1200", "gfx1201",
366366
#endif
367367
#if ROCM_VERSION >= 60500
368368
"gfx950"

aten/src/ATen/native/cuda/Blas.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -257,9 +257,9 @@ static bool isSupportedHipLtROCmArch(int index) {
257257
hipDeviceProp_t* prop = at::cuda::getDeviceProperties(index);
258258
std::string device_arch = prop->gcnArchName;
259259
static const std::vector<std::string> archs = {
260-
"gfx90a", "gfx942"
260+
"gfx90a", "gfx942",
261261
#if ROCM_VERSION >= 60300
262-
, "gfx1200", "gfx1201"
262+
"gfx1200", "gfx1201",
263263
#endif
264264
#if ROCM_VERSION >= 60500
265265
"gfx950"
@@ -934,7 +934,7 @@ static bool _scaled_mm_allowed_device() {
934934
static const std::vector<std::string> archs = {
935935
"gfx942",
936936
#if ROCM_VERSION >= 60300
937-
"gfx1200", "gfx1201"
937+
"gfx1200", "gfx1201",
938938
#endif
939939
#if ROCM_VERSION >= 60500
940940
"gfx950"

0 commit comments

Comments
 (0)