1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import os
import unittest
from .common import parse_args, run
from .torchbench import setup_torchbench_cwd, TorchBenchmarkRunner
try:
# fbcode only
from aiplatform.utils.sanitizer_status import is_asan_or_tsan
except ImportError:
def is_asan_or_tsan():
return False
class TestDynamoBenchmark(unittest.TestCase):
@unittest.skipIf(is_asan_or_tsan(), "ASAN/TSAN not supported")
def test_benchmark_infra_runs(self) -> None:
"""
Basic smoke test that TorchBench runs.
This test is mainly meant to check that our setup in fbcode
doesn't break.
If you see a failure here related to missing CPP headers, then
you likely need to update the resources list in:
//caffe2:inductor
"""
original_dir = setup_torchbench_cwd()
try:
args = parse_args(
[
"-dcpu",
"--inductor",
"--training",
"--performance",
"--only=BERT_pytorch",
"-n1",
"--batch-size=1",
]
)
run(TorchBenchmarkRunner(), args, original_dir)
finally:
os.chdir(original_dir)
|
