Skip to content

Commit 1e22aac

Browse files
authored
Use nvfuser_direct for version checking (#2812)
1 parent 406cd2f commit 1e22aac

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎scripts/validate_build.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def main():
1212
actual = os.path.abspath(torch.__file__)
1313
assert expected == actual, f"{expected} vs. {actual}"
1414

15-
import nvfuser
15+
import nvfuser_direct as nvfuser
1616
from looseversion import LooseVersion
1717

1818
assert LooseVersion(nvfuser.version()) >= LooseVersion("0.0.6"), nvfuser.version()

0 commit comments

Comments
 (0)