اصلاح کد ضرب ماتریسی با استفاده از tensor core

پروژه مرتبط با ضرب ماتریسی با استفاده از tensor core (کتابخونه wmma) می باشد. سه mode ضرب مختلف وجود دارد. تابع هر سه mode بدون tensor core نوشته شده است.  Mode 3 با tensor core  هم نوشته شده که به درستی کار می کنه و خروجی مشابه با نسخه بدون tensor core داره. تابع Mode 1 هم نوشته شده ولی خروجی اون اشتباهه و برابر با حالتی که از tensor core استفاده نمیکنه نیست. تابع Mode 2 خیلی به Mode 1 شبیه هست. تابع Mode 1 باید اصلاح بشه و Mode 2 هم باید نوشته بشه. اسلایدی در مورد نحوه کارکرد tensor core ها هم دارم و در صورت نیاز می تونم توضیحش رو هم بدهم.