IBM / pytorch-large-model-support

Large Model Support in PyTorch
Apache License 2.0
132 stars 19 forks source link

LMS for Nvidia A100 #13

Open junaidjawaid1 opened 1 year ago

junaidjawaid1 commented 1 year ago

I read your post of PyTorch issue #35633. it is mentioned "I am basically creating this issue because I really like LMS. So far I have waited the support of LMS for each version of PyTorch. Each time I had to manually compile PyTorch (and create wheels) to have the support of it" I was just curious if I can get any guidance as to how to that, and if there is any way to make LMS work with Nvidia A100

junaidjawaid1 commented 1 year ago

@UlrichFreitag I would really appreciate your comments on this. Thank you!