tests/test_tensor/test_hybrid_device.py · 0653c63eaacd1504f5d66f2e11f80defdb155832 · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[Tensor] 1d row embedding (#1075) · 0653c63e
Ziyue Jiang authored Jun 08, 2022
```
* Add CPU 1d row embedding

* polish
```
0653c63e

test_hybrid_device.py 3.13 KB