Skip to content

Commit 07752a9

Browse files
committed
[Lint]
1 parent 49e54af commit 07752a9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

examples/gemm/example_gemm_autotune.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,7 @@ def ref_program(A, B):
1616

1717
def get_configs(M, N, K, with_roller=False, topk=20):
1818
if with_roller:
19-
arch = CDNA("hip") if torch.version.hip is not None else CUDA("cuda")
20-
19+
arch = CUDA("cuda") if torch.version.hip is None else CDNA("hip")
2120
carve_template = MatmulTemplate(
2221
M=M,
2322
N=N,

0 commit comments

Comments
 (0)