Skip to content

Commit d7f9b54

Browse files
committed
Fix parameterSweeps scripts
1 parent 7f30da1 commit d7f9b54

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mlir/utils/performance/parameterSweeps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from typing import Callable, Iterable, List, Sequence, Optional, Tuple, TypeVar, Union
2222

2323
import perfRunner
24-
from perfRunner import Paths, ConvConfiguration, getArch, getNumCU
24+
from perfRunner import Paths, ConvConfiguration, getArch, getNumCU, getChip
2525
from perfCommonUtils import CORRECT_RESULT_RE
2626

2727
@dataclass(frozen=True)
@@ -525,7 +525,7 @@ def main() -> bool:
525525
print(f"""Unknown arch {arch}""", file=sys.stderr)
526526

527527
options = Options(debug=args.debug, quiet=args.quiet,
528-
arch=arch, flags=rocmlir_gen_flags, concurrent_tests=args.jobs, numCU=getNumCU(arch))
528+
arch=arch, flags=rocmlir_gen_flags, concurrent_tests=args.jobs, numCU=getNumCU(getChip()))
529529
paths = perfRunner.create_paths(None, args.mlir_build_dir)
530530

531531
config = args.config

0 commit comments

Comments
 (0)