search

iskyo0ps / compiler101_gpt_generated

Apache License 2.0
0 stars 0 forks source link

SWDEV-00002 - tune tvm #1

Closed iskyo0ps closed 3 months ago

iskyo0ps commented 3 months ago

we got a bad result when using tvm optimize model on commit id https://github.com/iskyo0ps/complier101_gpt_generated/commit/7aa28ae2a06d2a5a13dd47a7a83935351cbc1f8f Relay top-1 id: 281, class name: tabby, tabby cat Torch top-1 id: 281, class name: tabby, tabby cat Relay time: 0.41773597050000005 seconds Torch time: 0.09253386350000029 seconds

iskyo0ps commented 3 months ago

One or more operators have not been tuned. Please tune your model for better performance. Use DEBUG logging level to see more details.

iskyo0ps commented 3 months ago

Install a specific version of xgboost that is known to work with TVM(1.0.0), such as xgboost==1.3.3. pip install xgboost==1.3.3