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

Method with_scope_switch

python/tvm/tirx/exec_context.py:406–408  ·  view source on GitHub ↗
(self, scope_kind: str)

Source from the content-addressed store, hash-verified

404 )
405
406 def with_scope_switch(self, scope_kind: str) -> ExecContext:
407 split = scope_switch(self.A, scope_kind)
408 return ExecContext(A=self.A, scope_kind=scope_kind, inter=split.inter, intra=split.intra)

Callers 10

test_ex_8_1_cta_wg_warpFunction · 0.80
test_ex_8_2_filter_then_warpgroupFunction · 0.80
test_ex_8_3_sugar_warp_rangeFunction · 0.80
test_ex_8_4_widen_warp_to_wgFunction · 0.80
test_ex_8_5_partial_wgFunction · 0.80
test_ex_8_6_factor_failFunction · 0.80
test_ex_8_7_deep_nestedFunction · 0.80
test_ex_8_8_fa4_patternFunction · 0.80
test_ex_8_9_cross_cta_clusterFunction · 0.80
test_filter_is_pureFunction · 0.80

Calls 2

scope_switchFunction · 0.85
ExecContextClass · 0.85

Tested by 10

test_ex_8_1_cta_wg_warpFunction · 0.64
test_ex_8_2_filter_then_warpgroupFunction · 0.64
test_ex_8_3_sugar_warp_rangeFunction · 0.64
test_ex_8_4_widen_warp_to_wgFunction · 0.64
test_ex_8_5_partial_wgFunction · 0.64
test_ex_8_6_factor_failFunction · 0.64
test_ex_8_7_deep_nestedFunction · 0.64
test_ex_8_8_fa4_patternFunction · 0.64
test_ex_8_9_cross_cta_clusterFunction · 0.64
test_filter_is_pureFunction · 0.64