DeepSpeed
fa073576 - Fix the tensor-slicing with multi-GPU inference and kernel-injection (#1724)

Commit

3 years ago

Fix the tensor-slicing with multi-GPU inference and kernel-injection (#1724) * use the right tensor-copy function when adding tensor-slicing * small fix in inference tutorial

References

#1724 - Fix the tensor-slicing with multi-GPU inference and kernel-injection

Author

RezaYazdaniAminabadi

Parents

bea701a1

DeepSpeed fa073576 - Fix the tensor-slicing with multi-GPU inference and kernel-injection (#1724)

DeepSpeed
fa073576 - Fix the tensor-slicing with multi-GPU inference and kernel-injection (#1724)