PyTorch Operator Micro-benchmarks

Features

Getting Started

Initial Setup

How to run the benchmarks:

Code to support `torch.add` in the benchmark

Output and Command Line Control of the Benchmark

Command-Line Control

Adding New Operators to the Benchmark Suite

Add a New PyTorch Operator

Part 1. Specify Inputs to Operators

Part 2. Create Tensors and Add Computation

Part 3. Register Tests With the Benchmark Suite

Part 4. Run the Registered Tests

Add a List of Operators

Part 1. Specify the List of Operators

Part 2. Create Tensors and Add Computation

Part 3. Register a List of Operators

Add Gradient Ops

For PyTorch Gradient Ops