triton.language.extra.cann.extension.register_custom_op

triton.language.extra.cann.extension.register_custom_op(op)

Register a custom operation so that we can invoke it using al.custom().