Searched refs:benchmark_torch_function_in_microseconds (Results 1 – 4 of 4) sorted by relevance
192 def benchmark_torch_function_in_microseconds(f, *args, **kwargs): function239 nn_mha_time = benchmark_torch_function_in_microseconds(242 composite_mha_time = benchmark_torch_function_in_microseconds(251 compiled_nn_mha_time = benchmark_torch_function_in_microseconds(255 compiled_composite_mha_time = benchmark_torch_function_in_microseconds(
16 def benchmark_torch_function_in_microseconds(func: Callable, *args, **kwargs) -> float: function100 forward_time = benchmark_torch_function_in_microseconds(112 backward_time = benchmark_torch_function_in_microseconds(
31 def benchmark_torch_function_in_microseconds(func: Callable, *args, **kwargs) -> float: function178 forward_eager_time = benchmark_torch_function_in_microseconds(181 forward_compiled_time = benchmark_torch_function_in_microseconds(207 backward_eager_time = benchmark_torch_function_in_microseconds(220 backward_compile_time = benchmark_torch_function_in_microseconds(
18 def benchmark_torch_function_in_microseconds(func: Callable, *args, **kwargs) -> float: function162 materialized_mask_time = benchmark_torch_function_in_microseconds(165 attn_mask_subclass_time = benchmark_torch_function_in_microseconds(