frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-12 14:54:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix AI-PEP path error (#19514)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19514

as title

Reviewed By: hl475

Differential Revision: D15018499

fbshipit-source-id: 9ce38e3a577432e0575a6743f5dcd2e907d3ab9d
This commit is contained in:
Mingzhe Li
2019-04-19 16:22:12 -07:00
committed by Facebook Github Bot
parent a421f882dc
commit 5da7b74d48
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
benchmarks/operator_benchmark/ops/benchmark_all_test.py
View File

@ -3,14 +3,11 @@ from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import importlib
import os
from benchmarks.operator_benchmark import benchmark_runner
from benchmarks.operator_benchmark.ops import ( # noqa
add_test, # noqa
matmul_test) # noqa
if __name__ == "__main__":
# TODO: current way of importing other tests are fragile, so we need to have a robust way
for module in os.listdir(os.path.dirname(__file__)):
if module == '__init__.py' or not module.endswith('_test.py'):
continue
importlib.import_module("benchmarks.operator_benchmark.ops." + module[:-3])
benchmark_runner.main()