Skip to content

fix: skip pin_memory and non_blocking transfer for tensor subclasses in group offloading#13305

Open
s-zx wants to merge 1 commit intohuggingface:mainfrom
s-zx:fix/group-offloading-torchao-subclass-v2
Open

fix: skip pin_memory and non_blocking transfer for tensor subclasses in group offloading#13305
s-zx wants to merge 1 commit intohuggingface:mainfrom
s-zx:fix/group-offloading-torchao-subclass-v2

Commits

Commits on Mar 22, 2026