search

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
Apache License 2.0
416 stars 53 forks source link

ttnn implementation for yolox_m #8591

Open saichandax opened 4 months ago

saichandax commented 4 months ago

reference implementation: https://github.com/Megvii-BaseDetection/YOLOX/tree/main

ankitmcw commented 4 months ago

Yolox_m current status

  1. Reference implementation of yolox_m completed
  2. Tested reference against torch: Getting pcc 1 in all three output tensors
  3. Reference vs torch graph comparison checked
  4. Currently working on ttnn implementation

Torch graph: Yolox_m_torch.gv.pdf

Reference graph: Yolox_m_reference.gv.pdf

ankitmcw commented 4 months ago

Profile perf results:

Focus submodule: ops_perf_results_yolox_focus_2024_06_03_13_31_07.csv

Dark2 submodule: ops_perf_results_yolox_dark2_2024_06_03_13_09_36.csv

Dark3 submodule: ops_perf_results_yolox_dark3_2024_06_03_13_12_49.csv

Dark4 submodule:ops_perf_results_yolox_dark4_2024_06_03_13_13_36.csv

Dark5 submodule:ops_perf_results_yolox_dark5_2024_06_03_13_30_19.csv

CSPDarknet submodule: ops_perf_results_yolox_cspdarknet_2024_06_03_13_35_28.csv