[Drivers] Suppress OSError (#278)

hidet-org · Jun 13, 2023 · cdd75ec · cdd75ec
1 parent 09463e8
commit cdd75ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/hidet/drivers/build_module.py b/python/hidet/drivers/build_module.py
@@ -112,7 +112,10 @@ def build_job(args):
     if num_workers > 1 and len(jobs) > 1:
         # Set the affinity of current process. Some package such as numpy will change affinity of current process,
         # which might limit the parallelism of compilation.
-        os.sched_setaffinity(0, range(cpu_count))
+        from contextlib import suppress
+
+        with suppress(OSError):
+            os.sched_setaffinity(0, range(cpu_count))
 
         lazy_initialize_cuda()