1 parent 406cd2f commit 1e22aacCopy full SHA for 1e22aac
1 file changed
scripts/validate_build.py
@@ -12,7 +12,7 @@ def main():
12
actual = os.path.abspath(torch.__file__)
13
assert expected == actual, f"{expected} vs. {actual}"
14
15
- import nvfuser
+ import nvfuser_direct as nvfuser
16
from looseversion import LooseVersion
17
18
assert LooseVersion(nvfuser.version()) >= LooseVersion("0.0.6"), nvfuser.version()
0 commit comments