Test backward pass in unit test.

Summary:
In unit test, run both forward and backward pass.

If the backward pass throws `NotImplementedError`, skip the test since the operator does not support backward pass.

Reviewed By: int3

Differential Revision: D64471087

fbshipit-source-id: c9d0c43544314fc11305f271e8e80f7ba07b2675

Author: xuzhao9

torchbenchmark/util/triton_op.py

@@ -491,7 +491,7 @@ class BenchmarkOperator(metaclass=PostInitProcessor):
     _input_iter: Optional[Generator] = None
     extra_args: List[str] = []
     example_inputs: Any = None
-    use_cuda_graphs: bool = True
+    use_cuda_graphs: bool = False
 
     """
     A base class for adding operators to torch benchmark.