Skip to content

Commit

Permalink
Use 'old' instead of 'legacy' when specifying the fuser.
Browse files Browse the repository at this point in the history
  • Loading branch information
Mikhail Zolotukhin committed Sep 4, 2020
1 parent 26dbd26 commit 3b6cfaa
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions compare.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#!/bin/bash

pytest test_bench.py -k cuda-jit --fuser legacy --benchmark-json legacy.json
pytest test_bench.py -k cuda-jit --fuser old --benchmark-json old.json
pytest test_bench.py -k cuda-jit --fuser te --benchmark-json te.json
python compare.py legacy.json te.json
python compare.py old.json te.json
2 changes: 1 addition & 1 deletion conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ def pytest_addoption(parser):
parser.addoption("--fuser", help="fuser to use for benchmarks")

def set_fuser(fuser):
if fuser == "legacy":
if fuser == "old":
torch._C._jit_set_profiling_executor(False)
torch._C._jit_set_profiling_mode(False)
torch._C._jit_override_can_fuse_on_gpu(True)
Expand Down

0 comments on commit 3b6cfaa

Please sign in to comment.