提交 e0c78e29 编写于 作者: HansBug's avatar HansBug 😆

dev(narugo): torch.compile not support py3.11 yet

上级 88705f32
from unittest import skipUnless
import pytest
from hbutils.testing import vpip, OS
from hbutils.testing import vpip, OS, vpython
from treevalue import FastTreeValue, register_for_torch
......@@ -62,7 +62,7 @@ class TestTreeIntegrationTorch:
with pytest.warns(UserWarning):
register_for_torch(MyTreeValueX)
@skipUnless(vpip('torch') >= '2.0.0' and OS.linux, 'Torch 2 on linux platform required')
@skipUnless(vpip('torch') >= '2.0.0' and OS.linux and vpython < '3.11', 'Torch 2 on linux platform required')
def test_torch_compile(self):
@torch.compile
def foo(x, y, t):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册