https://tvm.apache.org/docs/dev/relay_op_strategy.html is a tutorial that covers a lot of the strategy framework. If this is an existing operator, you should be fine just adding strategy.add_implementation
to op/strategy/cuda.py
.
https://tvm.apache.org/docs/dev/relay_op_strategy.html is a tutorial that covers a lot of the strategy framework. If this is an existing operator, you should be fine just adding strategy.add_implementation
to op/strategy/cuda.py
.