Skip to content

Commit 345b9db

Browse files
committed
- fix unit test
1 parent 6c965d7 commit 345b9db

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

tests/python/relay/transform/test_compiler_function_utils.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,7 @@ def make_consts(dtype, shapes):
3838
(2304,), # 1
3939
(600, 32, 64), # 2
4040
],
41-
),
42-
"attributes": [{"relay_attrs": None}],
41+
)
4342
}
4443

4544

@@ -115,7 +114,7 @@ def expected_extern_mod():
115114
"""
116115
#[version = "0.0.5"]
117116
def @main(%x0 : Tensor[(1600, 768), float16], %x3 : Tensor[(600, 32, 64), float16]) -> (Tensor[(1600, 2304), float16], Tensor[(600, 32, 32), float16]) {
118-
%1 = call_lowered(@tvmgen_default_cutlass_main_0, (%x0, meta[relay.Constant][0], meta[relay.Constant][1]), metadata=meta[attributes][0]);
117+
%1 = @tvmgen_default_cutlass_main_0(%x0, meta[relay.Constant][0], meta[relay.Constant][1]);
119118
%2 = fn(%y_3_i0: Tensor[(600, 32, 64), float16], %y_3_i1: Tensor[(600, 32, 64), float16],
120119
Inline=1, Compiler="cublas", global_symbol="tvmgen_default_cublas_main_3", Primitive=1) -> Tensor[(600, 32, 32), float16] {
121120
%6 = fn (%FunctionVar_0_01: Tensor[(600, 32, 64), float16], %FunctionVar_0_11: Tensor[(600, 32, 64), float16],

0 commit comments

Comments
 (0)