Skip to content

Commit 9f45950

Browse files
committed
skip for rmsnorm
1 parent 04696bf commit 9f45950

6 files changed

+18
-0
lines changed

tools/pnnx/tests/ncnn/test_transformers_deepseek_v3_attention.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ def forward(self, x, mask0):
2929
return out0[0]
3030

3131
def test():
32+
if version.parse(torch.__version__) < version.parse('2.4'):
33+
return True
34+
3235
net = Model()
3336
net.eval()
3437

tools/pnnx/tests/ncnn/test_transformers_qwen2_attention.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ def forward(self, x, mask0):
2929
return out0[0]
3030

3131
def test():
32+
if version.parse(torch.__version__) < version.parse('2.4'):
33+
return True
34+
3235
net = Model()
3336
net.eval()
3437

tools/pnnx/tests/ncnn/test_transformers_qwen3_attention.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ def forward(self, x, mask0):
2929
return out0[0]
3030

3131
def test():
32+
if version.parse(torch.__version__) < version.parse('2.4'):
33+
return True
34+
3235
net = Model()
3336
net.eval()
3437

tools/pnnx/tests/test_transformers_deepseek_v3_attention.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ def forward(self, x, mask0):
2929
return out0[0]
3030

3131
def test():
32+
if version.parse(torch.__version__) < version.parse('2.4'):
33+
return True
34+
3235
net = Model()
3336
net.eval()
3437

tools/pnnx/tests/test_transformers_qwen2_attention.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ def forward(self, x, mask0):
2929
return out0[0]
3030

3131
def test():
32+
if version.parse(torch.__version__) < version.parse('2.4'):
33+
return True
34+
3235
net = Model()
3336
net.eval()
3437

tools/pnnx/tests/test_transformers_qwen3_attention.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@ def forward(self, x, mask0):
2929
return out0[0]
3030

3131
def test():
32+
if version.parse(torch.__version__) < version.parse('2.4'):
33+
return True
34+
3235
net = Model()
3336
net.eval()
3437

0 commit comments

Comments
 (0)