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

Method _get_cached_method

python/tvm/runtime/disco/session.py:119–128  ·  view source on GitHub ↗
(self, name: str)

Source from the content-addressed store, hash-verified

117 __slots__ = ("_cache", "_import_python_module")
118
119 def _get_cached_method(self, name: str) -> Callable:
120 if not hasattr(self, "_cache"):
121 cache = self._cache = {} # pylint: disable=attribute-defined-outside-init
122 else:
123 cache = self._cache
124 if name not in cache:
125 func = cache[name] = self.get_global_func(name)
126 else:
127 func = cache[name]
128 return func
129
130 def empty(
131 self,

Callers 8

emptyMethod · 0.95
load_vm_moduleMethod · 0.95
scatter_from_worker0Method · 0.95
gather_to_worker0Method · 0.95
allreduceMethod · 0.95
allgatherMethod · 0.95
__getitem__Method · 0.80

Calls 1

get_global_funcMethod · 0.95

Tested by

no test coverage detected