search

tenstorrent / tt-tvm

TVM for Tenstorrent ASICs
Apache License 2.0
18 stars 6 forks source link

Register strategy for pixel shuffle #6

Closed meenakshiramanathan1 closed 3 months ago

meenakshiramanathan1 commented 3 months ago

Model failed in TVM compilation with the following error

AssertionError: pixel_shuffle doesn't have an FTVMStrategy registered. You can register one in python with `tvm.relay.op.register_strategy