Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dcollection.cpp1563 std::function<bool()>& cuda_sync_enabled_fn() { in cuda_sync_enabled_fn() function
1570 return cuda_sync_enabled_fn()(); in get_cuda_sync_enabled()
1574 cuda_sync_enabled_fn() = std::move(fn); in set_cuda_sync_enabled_fn()
1578 cuda_sync_enabled_fn() = [val]() { return val; }; in set_cuda_sync_enabled_val()