intel-extension-for-pytorch

1
import argparse
2
import intel_extension_for_pytorch as ipex
3

4

5
def create_cpu_pool(args):
6
    core_ids = [1, 2]
7
    cpu_pool = ipex.cpu.runtime.CPUPool(core_ids)
8
    print("The created CPUPool has core is: {}".format(cpu_pool.core_ids), flush=True)
9

10

11
if __name__ == "__main__":
12
    parser = argparse.ArgumentParser()
13
    parser.add_argument("--case-name", default="create_cpu_pool", type=str)
14
    args = parser.parse_args()
15
    if args.case_name == "create_cpu_pool":
16
        create_cpu_pool(args)
17