MCPcopy Index your code
hub / github.com/apache/tvm / func_a

Function func_a

tests/python/relax/test_utils.py:219–234  ·  view source on GitHub ↗
(n: R.Prim("int64"))

Source from the content-addressed store, hash-verified

217
218 @R.function(private=True)
219 def func_a(n: R.Prim("int64")):
220 @R.function
221 def recursive_lambda(i_arg: R.Prim(value="i")) -> R.Prim("int64"):
222 i = T.int64()
223 if R.prim_value(i == 0):
224 output = R.prim_value(T.int64(0))
225 # ^
226 # The first mismatch is here ^
227 else:
228 remainder_relax = recursive_lambda(R.prim_value(i - 1))
229 remainder_tir = T.int64()
230 _ = R.match_cast(remainder_relax, R.Prim(value=remainder_tir))
231 output = R.prim_value(i + remainder_tir)
232 return output
233
234 return recursive_lambda(n)
235
236 @R.function(private=True)
237 def func_b(n: R.Prim("int64")):

Callers

nothing calls this directly

Calls 1

recursive_lambdaFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…