[fbcode] Fix operator_benchmark with jit mode (#67382)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67382
two simple updates:
* fix running benchmark with --use_jit. Previously will fail with error
torch.jit.frontend.UnsupportedNodeError: import statements aren't supported:
File "/proc/self/fd/3/bmm_test.py", line 9
def __invoke_main():
import ctypes
~~~~~~ <--- HERE
import ctypes.util
import errno
* add matmul to bmm benchmark as D31837588
Test Plan:
buck run mode/opt caffe2/benchmarks/operator_benchmark/pt:bmm_test -- --forward_only=True --mkl_num_threads=1 --omp_num_threads=1
--use_jit=True
Reviewed By: ShijunK
Differential Revision: D31960528
fbshipit-source-id: 84b892934149784d1b8a0f90b0233cc2f1cf1f5f
