1from pt import ( # noqa: F401 2 add_test, 3 ao_sparsifier_test, 4 as_strided_test, 5 batchnorm_test, 6 binary_test, 7 cat_test, 8 channel_shuffle_test, 9 chunk_test, 10 conv_test, 11 diag_test, 12 embeddingbag_test, 13 fill_test, 14 gather_test, 15 groupnorm_test, 16 hardsigmoid_test, 17 hardswish_test, 18 instancenorm_test, 19 interpolate_test, 20 layernorm_test, 21 linear_test, 22 matmul_test, 23 nan_to_num_test, 24 pool_test, 25 remainder_test, 26 softmax_test, 27 split_test, 28 sum_test, 29 tensor_to_test, 30) 31 32import operator_benchmark as op_bench 33 34 35if __name__ == "__main__": 36 op_bench.benchmark_runner.main() 37