[JIT] Add IR Benchmarking tests to ai bench (#35732)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35732
Adding IR Complexity benchmarking tests to AI Bench. For full description of IR benchmarking look at https://github.com/pytorch/pytorch/pull/34918.
Test Plan:
https://our.intern.facebook.com/intern/testinfra/testconsole/testrun/5348024580589132/ test run
Local run:
PyTorchObserver {"type": "NET", "metric": "conv1d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv1d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv2d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv2d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv3d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv3d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_transpose1d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_transpose1d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_transpose2d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_transpose2d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_transpose3d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_transpose3d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_tbc_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "conv_tbc_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "avg_pool1d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "avg_pool1d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "avg_pool2d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "avg_pool2d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "avg_pool3d_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "avg_pool3d_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
I0403 15:39:29.672143 2015194 init.cc:579] Skip logging in unit test environment for event: torch.script.compile
PyTorchObserver {"type": "NET", "metric": "Linear_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Linear_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Threshold_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Threshold_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "ReLU_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "ReLU_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "ReLU6_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "ReLU6_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "RReLU_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "RReLU_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Hardtanh_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Hardtanh_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Sigmoid_profiled_num_ifs_loops", "unit": "scalar", "value": "0.0"}
PyTorchObserver {"type": "NET", "metric": "Sigmoid_profiled_num_non_tensor_nodes", "unit": "scalar", "value": "0.0"}
Reviewed By: driazati
Differential Revision: D20754238
fbshipit-source-id: 179240dee516647e5583b9fe47083c84241ddacc