onnxruntime
CUDA kernel for ClipGradNorm for TensorSeq gradients
#12412
Merged

CUDA kernel for ClipGradNorm for TensorSeq gradients #12412

baijumeswani merged 6 commits into master from bmeswani/clipgradnorm
baijumeswani
baijumeswani baijumeswani added training
baijumeswani baijumeswani requested a review from askhade askhade 3 years ago
baijumeswani baijumeswani requested a review from pengwa pengwa 3 years ago
baijumeswani baijumeswani requested a review from ashbhandare ashbhandare 3 years ago
baijumeswani Add CUDA kernel for ClipGradNorm for TensorSeq gradients
293a7e56
baijumeswani Add support for TensorSeq for graph outputs in InferenceSession
a1119e17
baijumeswani baijumeswani force pushed from 1eb026f4 to a1119e17 3 years ago
baijumeswani Use inlined containers
770466f9
pengwa
pengwa commented on 2022-08-03
baijumeswani Use ClipGradNormInplace to indicate inplace update and resolve other …
6c9cb32e
askhade
askhade commented on 2022-08-03
askhade
askhade commented on 2022-08-03
askhade
askhade commented on 2022-08-03
askhade
askhade commented on 2022-08-03
askhade
askhade commented on 2022-08-03
askhade
askhade commented on 2022-08-03
baijumeswani Change name to InplaceClipGradNorm and resolve other pull request rev…
b91946fc
baijumeswani baijumeswani force pushed from 536797d5 to b91946fc 3 years ago
askhade
askhade commented on 2022-08-04
askhade
askhade dismissed these changes on 2022-08-04
baijumeswani Move norm type assertion and resolve other pull request review comments
9cd2cdae
baijumeswani baijumeswani dismissed their stale review via 9cd2cdae 3 years ago
pengwa
pengwa approved these changes on 2022-08-05
baijumeswani baijumeswani merged a7d62907 into master 3 years ago
baijumeswani baijumeswani deleted the bmeswani/clipgradnorm branch 3 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone