اصلاح کد ضرب ماتریسی با استفاده از 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 ها هم دارم و در صورت نیاز می تونم توضیحش رو هم بدهم.

1 پیشنهاد 20,000,000 تومان 3 روز پیش

مهارت ها