DeepSpeed
61e07786 - Added wrappers for hpu tensors based on dtype (#5771)

Commit
1 year ago
Added wrappers for hpu tensors based on dtype (#5771) This avoids graph breaks when using torch.compile.
Author
Parents
Loading