Skip to content

Commit 1b9bad8

Browse files
committed
review comment
1 parent 7070ce9 commit 1b9bad8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/ingress/convert-kernel-bench-to-mlir.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
mlir_kernels_dir = project_root / "cache" / "ingress" / "KernelBench"
2424
free_mem_gb = psutil.virtual_memory().available // (1024**3)
2525
print(f"Available memory: {free_mem_gb} GB")
26-
max_workers = free_mem_gb // 12 # some workers need 12~16GB
26+
max_workers = min(free_mem_gb // 12, psutil.cpu_count()) # some workers need 12~16GB
2727
print(f"Using max_workers={max_workers} based on available memory")
2828

2929
if not torch_kernels_dir.is_dir():

0 commit comments

Comments
 (0)