builder icon indicating copy to clipboard operation
builder copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Update sm90 to -> sm90a

Open drisspg opened this issue 8 months ago • 3 comments

Summary

Sm90a is needed to enable some features found in Cutlass. For some reason google as the worst SEO for finding more information about this gencode. This is the best description I have found: https://github.com/NVIDIA/cccl/issues/1270

I am not sure if we want to instead build for sm90 and sm90a ?

Pytorch Features requiring this change: https://github.com/pytorch/pytorch/pull/128989

cc @ptrblck @nWEIdia

drisspg avatar Jun 20 '24 17:06 drisspg