Skip to content

Commit

Permalink
skip asm check for askew weight shapes
Browse files Browse the repository at this point in the history
  • Loading branch information
Valery Chernov committed Dec 23, 2022
1 parent 7c6a6a2 commit 5601b8d
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 6 deletions.
7 changes: 4 additions & 3 deletions tests/python/relay/test_op_level1.py
Original file line number Diff line number Diff line change
Expand Up @@ -815,9 +815,10 @@ def test_dense_skylake_avx512(m, n, k):
with tvm.transform.PassContext(opt_level=3):
lib = relay.build(mod, target=target)

asm = lib.lib.get_source("asm")
assert "pmaddubs" in asm
assert "pmaddw" in asm
if n%16 == 0 and k%4 == 0:
asm = lib.lib.get_source("asm")
assert "pmaddubs" in asm
assert "pmaddw" in asm

dev = tvm.device(target, 0)
runtime = tvm.contrib.graph_executor.GraphModule(lib["default"](dev))
Expand Down
7 changes: 4 additions & 3 deletions tests/python/relay/test_op_level10.py
Original file line number Diff line number Diff line change
Expand Up @@ -547,9 +547,10 @@ def test_batch_matmul_skylake_avx512(b, m, n, k):
with tvm.transform.PassContext(opt_level=3):
lib = relay.build(mod, target=target)

asm = lib.lib.get_source("asm")
assert "pmaddubs" in asm
assert "pmaddw" in asm
if n%16 == 0 and k%4 == 0:
asm = lib.lib.get_source("asm")
assert "pmaddubs" in asm
assert "pmaddw" in asm

dev = tvm.device(target, 0)
runtime = tvm.contrib.graph_executor.GraphModule(lib["default"](dev))
Expand Down

0 comments on commit 5601b8d

Please sign in to comment.