tvm_tirx.backend.cuda.tile.exec_context.filter_modulo

Contents

tvm_tirx.backend.cuda.tile.exec_context.filter_modulo#

tvm_tirx.backend.cuda.tile.exec_context.filter_modulo(A, axis, modulus, residue)#

Intersect an active-set axis with axis % modulus == residue.

Parameters:
  • A (ActiveSet)

  • axis (str)

  • modulus (int)

  • residue (int)

Return type:

ActiveSet