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

Function test_let_var

tests/python/tirx/test_parser_printer.py:1230–1237  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

1228 # fmt: off
1229 @T.prim_func
1230 def test_let_var():
1231 T.device_entry()
1232 smem = T.alloc_shared([128], "float16")
1233 ptr: T.let[T.Var(name="ptr", dtype=PointerType(PrimType("uint64")))] = T.reinterpret(
1234 "handle", smem.access_ptr("rw")
1235 )
1236 T.evaluate(ptr)
1237 # fmt: on
1238 code = test_let_var.script()
1239 assert from_source(code).script() == code
1240

Callers

nothing calls this directly

Calls 4

PointerTypeClass · 0.90
PrimTypeClass · 0.90
access_ptrMethod · 0.80
evaluateMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…