How to specify the customized codegen when building the model? should it be something like this?
relay.build(mod, newCodeGen, params=params)
where the newCodeGen is my customized codegen.
Or should it be something like this based on the cutlass example
tvm.get_global_func("relay.ext.cutlass", True)