Apache TVM Discuss
How to move data-dependent such as unique and nonzeros on CPU and Other Op still run in GPU
Questions
dingyongchao
November 18, 2022, 8:51am
1
I have a model, such as
image
129×172 4.86 KB
And I want to place unique op and pre-op in CPU, and Other GEMM Op into GPU, How can I do it in TVM?