pytorch/torch/testing
Elias Ellison 4ced22c5de [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
2020-04-06 16:14:25 -07:00
..
_internal [JIT] Add IR Benchmarking tests to ai bench (#35732) 2020-04-06 16:14:25 -07:00
__init__.py Removes integer division call sites (#35862) 2020-04-02 17:04:15 -07:00