Searched refs:get_node_logical_cores (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/backends/xeon/ |
| H A D | test_launch.py | 48 assert cpuinfo.get_node_logical_cores(0) == [0, 1, 2, 3, 8, 9, 10, 11]
49 assert cpuinfo.get_node_logical_cores(1) == [4, 5, 6, 7, 12, 13, 14, 15]
|
| /aosp_15_r20/external/pytorch/torch/backends/xeon/ |
| H A D | run_cpu.py | 216 def get_node_logical_cores(self, node_id): member in _CPUinfo
462 cores = self.cpuinfo.get_node_logical_cores(args.node_id)
|