项目文件夹

文件

T

文件历史

nv-dlasalle 020f02498c [Performance][Optimizer] Enable using UVA and FP16 with SparseAdam Optimizer (#3885 )

* Add uva by default to embedding

* More updates

* Update optimizer

* Add new uva functions

* Expose new pinned memory function

* Add unit tests

* Update formatting

* Fix unit test

* Handle auto UVA case when training is on CPU

* Allow per-embedding decisions for whether to use UVA

* Address spares_optim.py comments

* Remove unused templates

* Update unit test

* Use dgl allocate memory for pinning

* allow automatically unpin

* workaround for d2h copy with a different dtype

* fix linting

* update error message

* update copyright

Co-authored-by: Xin Yao <xiny@nvidia.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>

2022-06-24 09:20:08 +08:00

mxnet

[Feature] Add support for sparse embedding (#2451 )

2021-01-28 00:26:49 +08:00

pytorch

[Performance][Optimizer] Enable using UVA and FP16 with SparseAdam Optimizer (#3885 )

2022-06-24 09:20:08 +08:00

tensorflow

[Feature] Add support for sparse embedding (#2451 )

2021-01-28 00:26:49 +08:00

__init__.py

[Feature] Add support for sparse embedding (#2451 )

2021-01-28 00:26:49 +08:00