Home
last modified time | relevance | path

Searched refs:privateuse1_device_handler (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/utils/benchmark/utils/
Dtimer.py21 privateuse1_device_handler = getattr(torch, torch._C._get_privateuse1_backend_name(), None) \ variable
25 if privateuse1_device_handler:
26 privateuse1_device_handler.synchronize()