We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7070ce9 commit 1b9bad8Copy full SHA for 1b9bad8
examples/ingress/convert-kernel-bench-to-mlir.py
@@ -23,7 +23,7 @@
23
mlir_kernels_dir = project_root / "cache" / "ingress" / "KernelBench"
24
free_mem_gb = psutil.virtual_memory().available // (1024**3)
25
print(f"Available memory: {free_mem_gb} GB")
26
-max_workers = free_mem_gb // 12 # some workers need 12~16GB
+max_workers = min(free_mem_gb // 12, psutil.cpu_count()) # some workers need 12~16GB
27
print(f"Using max_workers={max_workers} based on available memory")
28
29
if not torch_kernels_dir.is_dir():
0 commit comments