MCPcopy Create free account
hub / github.com/apache/tvm / test_gemm

Function test_gemm

tests/python/tirx/test_op.py:39–44  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

37
38
39def test_gemm():
40 A = decl_buffer((64, 64), "float32", scope="global")
41 B = decl_buffer((64, 64), "float32", scope="global")
42 C = decl_buffer((64, 64), "float32", scope="global")
43 D = decl_buffer((64, 64), "float32", scope="global")
44 _test("gemm", D[:, :], A[:, :], B[:, :], C[:, :], True, False, 1.0, 0.0)
45
46
47def test_buffer_replacer_no_shared_default():

Callers 1

test_op.pyFile · 0.70

Calls 2

decl_bufferFunction · 0.90
_testFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…