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

Method with_cta_axis_modulo

python/tvm/tirx/exec_context.py:399–404  ·  view source on GitHub ↗
(self, axis: str, modulus: int, residue: int)

Source from the content-addressed store, hash-verified

397 )
398
399 def with_cta_axis_modulo(self, axis: str, modulus: int, residue: int) -> ExecContext:
400 new_A = filter_modulo(self.A, axis, modulus, residue)
401 split = scope_switch(new_A, self.scope_kind)
402 return ExecContext(
403 A=new_A, scope_kind=self.scope_kind, inter=split.inter, intra=split.intra
404 )
405
406 def with_scope_switch(self, scope_kind: str) -> ExecContext:
407 split = scope_switch(self.A, scope_kind)

Callers

nothing calls this directly

Calls 3

filter_moduloFunction · 0.85
scope_switchFunction · 0.85
ExecContextClass · 0.85

Tested by

no test coverage detected