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

Function func_gen

python/tvm/tirx/script/builder/ir.py:2360–2368  ·  view source on GitHub ↗

Generate a DtypeConstructor for each PrimExpr dtype. Parameters ---------- name: str The ffi function name to call, e.g. "Float32", "Int32".

(name: str)

Source from the content-addressed store, hash-verified

2358
2359
2360def func_gen(name: str):
2361 """Generate a DtypeConstructor for each PrimExpr dtype.
2362
2363 Parameters
2364 ----------
2365 name: str
2366 The ffi function name to call, e.g. "Float32", "Int32".
2367 """
2368 return DtypeConstructor(name, _ffi_name_to_dtype(name))
2369
2370
2371def static_assert(x: Any, message: str = ""):

Callers 1

ir.pyFile · 0.85

Calls 2

DtypeConstructorClass · 0.85
_ffi_name_to_dtypeFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…