Searched defs:cur_stream (Results 1 – 3 of 3) sorted by relevance
62 c10::xpu::XPUStream cur_stream = c10::xpu::getCurrentXPUStream(); in TEST() local104 c10::xpu::XPUStream cur_stream = c10::xpu::getCurrentXPUStream(); in TEST() local
109 at::cuda::CUDAStream cur_stream = at::cuda::getCurrentCUDAStream(); in TEST() local
436 def fn(x, cur_stream, new_stream): argument