Fix build on cuda 12.5#22

keisukefukuda wants to merge 3 commits intofacebookresearch:mainfacebookresearch/dietgpu:mainfrom

keisukefukuda:fix-on-cuda-12.5keisukefukuda/dietgpu:fix-on-cuda-12.5Copy head branch name to clipboard