numa set local alloc

Signed-off-by: Mauricio J. Serrano <[email protected]>

Author: mserranos

aiu_fms_testing_utils/scripts/inference.py

@@ -343,6 +343,8 @@
             numa_rank = dist.get_rank()
             numa_node = dist.get_rank() // numa_size_per_node
             schedule.run_on_nodes(numa_node)
+            from numa import memory
+            memory.set_local_alloc()
             dprint(f"NUMA: process {numa_rank} set to node {numa_node}")
         except:
             dprint(f"NUMA not available in this machine, please install libnuma libraries")