Fix in benchmark_test_generator (#20237)

Summary: Add missing import Pull Request resolved: #20237 Differential Revision: D15245957 Pulled By: ilia-cher fbshipit-source-id: 0f71aa08eb9ecac32002a1644838d06ab9faa37c
pytorch · May 8, 2019 · eecf52b · eecf52b
1 parent 8a37518
commit eecf52b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/benchmarks/operator_benchmark/benchmark_test_generator.py b/benchmarks/operator_benchmark/benchmark_test_generator.py
@@ -8,6 +8,8 @@
 from operator_benchmark.benchmark_pytorch import PyTorchOperatorTestCase
 from operator_benchmark.benchmark_utils import * # noqa
 
+import torch
+
 
 def generate_test(configs, map_config, ops, OperatorTestCase):
     """