Skip to content

fix: use device-agnostic autocast and tensor placement for ROCm/XPU compatibility#15921

Open
zhihuidu-amd wants to merge 1 commit into
NVIDIA-NeMo:mainfrom
zhihuidu-amd:rocm/device-agnostic-autocast
Open

fix: use device-agnostic autocast and tensor placement for ROCm/XPU compatibility#15921
zhihuidu-amd wants to merge 1 commit into
NVIDIA-NeMo:mainfrom
zhihuidu-amd:rocm/device-agnostic-autocast

fix: use device-agnostic autocast and tensor placement for ROCm/XPU c…

ade5873
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jul 17, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO