From eecf52b444bac5e64dd0093d9fa0f0dc3aeff84c Mon Sep 17 00:00:00 2001
From: Ilia Cherniavskii <iliacher@fb.com>
Date: Tue, 7 May 2019 16:54:37 -0700
Subject: [PATCH] Fix in benchmark_test_generator (#20237)

Summary:
Add missing import
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20237

Differential Revision: D15245957

Pulled By: ilia-cher

fbshipit-source-id: 0f71aa08eb9ecac32002a1644838d06ab9faa37c
---
 benchmarks/operator_benchmark/benchmark_test_generator.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/benchmarks/operator_benchmark/benchmark_test_generator.py b/benchmarks/operator_benchmark/benchmark_test_generator.py
index 9ad3e722606..5b6e948e1f9 100644
--- a/benchmarks/operator_benchmark/benchmark_test_generator.py
+++ b/benchmarks/operator_benchmark/benchmark_test_generator.py
@@ -8,6 +8,8 @@ from operator_benchmark.benchmark_caffe2 import Caffe2OperatorTestCase
 from operator_benchmark.benchmark_pytorch import PyTorchOperatorTestCase
 from operator_benchmark.benchmark_utils import * # noqa
 
+import torch
+
 
 def generate_test(configs, map_config, ops, OperatorTestCase):
     """