tvm_tirx.backend.cuda.codegen.register_codegen

tvm_tirx.backend.cuda.codegen.register_codegen(op, backend='cuda')

Register a codegen function for a given op.

The codegen function should return a func_call statement, and optionally a list of tags that the codegen function needs.