Fix build on CUDA 12.6 and Pytorch >= 2.1#26

keisukefukuda wants to merge 1 commit intofacebookresearch:mainfacebookresearch/dietgpu:mainfrom

keisukefukuda:fix-build-cuda-12.6keisukefukuda/dietgpu:fix-build-cuda-12.6Copy head branch name to clipboard