Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Tags: yukirora/apex

Tags

rocm4.5.2

Toggle rocm4.5.2's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Update README.md

v0.3.3

Toggle v0.3.3's commit message
in multi tensor apply, skip empty tensors

v0.3.2

Toggle v0.3.2's commit message
Tag v0.3.2 works with PyTorch docker up to ROCm4.0, it includes sever…

…al critical bug fixes

v0.3.1

Toggle v0.3.1's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Fix LayerNorm op on ROCm (ROCm#36)

* fix warp size in WARP_SHFL* in layernorm

* enable fused_layer_norm tests on ROCm

v0.3

Toggle v0.3's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
update readme with ninja build instruction and pip3.6 install (ROCm#35)

v0.2

Toggle v0.2's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Merge pull request ROCm#18 from lcskrishna/cl/readme-updates

[Documentation] Update the readme for ROCm.

v0.1

Toggle v0.1's commit message

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
Enable support for sparse tensors for multi_tensor_apply (ROCm#6)