Searched defs:cuda_sync (Results 1 – 2 of 2) sorted by relevance
126 def run_jit_forward(self, num_runs, print_per_iter=False, cuda_sync=False): argument148 def run_forward(self, num_runs, print_per_iter, cuda_sync): argument
27 def cuda_sync(func, *args, **kwargs): function