frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix in benchmark_test_generator (#20237)

Browse Source 
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
This commit is contained in:
Ilia Cherniavskii
2019-05-07 16:54:37 -07:00
committed by Facebook Github Bot
parent 8a375189ea
commit eecf52b444
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/operator_benchmark/benchmark_test_generator.py
View File

@ -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):
"""