 Andrzej Kotłowski
|
7b1639f1aa
|
[TESTS] Make pytorch test_nn.py fully reproducible (#5887)
|
2023-06-26 09:51:31 +08:00 |
|
|
Hongzhi (Steve), Chen
|
98325b1097
|
[Misc] Black auto fix. (#4691)
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
|
2022-10-10 10:53:51 +08:00 |
|
|
Zihao Ye
|
bc209bff94
|
[bugfix] Fix the deterministic of kernel test and add documentation. (#1799)
* upd
* upd
* upd
* relax
* relax too
* upd
|
2020-07-14 04:22:31 +08:00 |
|
|
Da Zheng
|
cadcc1c2a6
|
[Feature] add sparse embedding. (#1497)
* add sparse embedding.
* fix
* add test.
* man fixes.
* many fixes
* fix sparse emb.
* fix.
* fix lint.
* fix lint.
* fix kvstore.
* expose DistTensor.
* test sparse embeddings.
* add attach_grad to the backends.
* remove part_id
* fix.
* move backward computation.
* move more computation to backend.
* fix a bug when applying learning rate.
* fix a few things.
* fix a few things.
* add docstring
* fix.
* apply no_grad.
* fix tests.
* fix for other frameworks.
* add examples in docstring.
|
2020-07-03 10:42:09 -07:00 |
|
|
Zihao Ye
|
071cba1f2b
|
[kernel] New SpMM & SDDMM kernel on CPU and CUDA (#1644)
* udp
* simplify
* sddmm dot cpu
* upd
* format
* upd
* compatible with MJ's PR
* lint
* upd
* upd
* upd
* python end
* upd
* upd
* lint
* lint
* upd
* upd
* upd
* upd
* upd
* lint
* fix mxnet
* upd
* lint
* use minjie's ptr
* macro
* upd
* reorg
* lint
* fix corner cases
* upd
* enrich cpu docs
* upd
* upd
* lint
* lint
* pylint
* sx review
* improve docstring
* python doc
* upd
* restructure
* lint
* upd test
* upd
* pylint
* fix corner cases and test
|
2020-06-22 23:18:32 +08:00 |
|
|
VoVAllen
|
a00636a02b
|
[Tensorflow] Several nn & example (#1191)
* several nn example
* appnp
* fix lint
* lint
* add dgi
* fix
* fix
* fix
* fff
* docs
* 111
* fix
* change init
* change result
* tiaocan+1
* fix
* fix lint
* fix
* fix
|
2020-01-19 22:48:40 +08:00 |
|
|
VoVAllen
|
d30a69bf27
|
[Backend] TF backend (#978)
* tf
* add builtin support
* fiix
* pytest
* fix
* fix
* fix some bugs
* fix selecting
* fix todo
* fix test
* fix test fail in tf
* fix
* fix
* fix gather row
* fix gather row
* log backend
* fix gather row
* fix gather row
* fix for pytorch
* fix
* fix
* fix
* fix
* fix
* fix tests
* fix
* fix
* fix
* fix
* fix
* fix
* fix convert
* fix
* fix
* fix
* fix inplace
* add alignment setting
* add debug option
* Revert "add alignment setting"
This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.
* tf ci
* fix lint
* fix lint
* add tfdlpack
* fix type
* add env
* fix backend
* fix
* fix tests
* remove one_hot
* remove comment
* remove comment
* fix
* use pip to install all
* fix test
* fix base
* fix
* fix
* add skip
* upgrade cmake
* change version
* change ci
* fix
* fix
* fix
* fix
* fix seg fault
* fix
* fix python version
* fix
* try fix
* fix
* fix
* tf takes longer time in ci
* change py version
* fix
* fix
* fix oom
* change kg env
* change kg env
* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊
* 我再也不搞各种乱七八糟环境了……
* use pytest
* Chang image
|
2019-12-20 15:56:51 +08:00 |
|