PR #10307 Improve NonZero on CUDA/ROCM

Improve NonZero on CUDA/ROCM #10307

pengwa merged 8 commits into master from pengwa/non_zero

improve NonZero

e0db3bac

pengwa requested a review from

weixingzhang 4 years ago

pengwa added training

pengwa requested a review from

mindest 4 years ago

fix megatron_fp16 optimzier, fix the doc

d3c649d8

pengwa requested a review from

baijumeswani 4 years ago

pengwa requested a review from

liqunfu 4 years ago

pengwa requested a review from

SherlockNoMad 4 years ago

pengwa requested a review from

thiagocrepaldi 4 years ago

pengwa requested a review from

tlh20 4 years ago

pengwa requested a review from

xadupre 4 years ago

multi_tensor_applier

2949e543

weixingzhang dismissed these changes on 2022-03-23

Merge branch 'master' of https://github.com/microsoft/onnxruntime int…

4d53fc36

mindest commented on 2022-03-24

mindest dismissed these changes on 2022-03-24

resolve comment

e1259e11

pengwa dismissed their stale review via e1259e11 4 years ago

fix building warning

5e333a88

fix build error when enabling training and use tensorrt

7f8f8b45

Merge branch 'master' of https://github.com/microsoft/onnxruntime int…

b03e792e

baijumeswani approved these changes on 2022-03-24

pengwa merged 89ef987a into master 4 years ago

pengwa deleted the pengwa/non_zero branch 4 years ago

Reviewers

baijumeswani

mindest

weixingzhang

liqunfu

SherlockNoMad

thiagocrepaldi

tlh20

xadupre

Assignees

No one assigned

Labels

training

Milestone

No milestone

onnxruntime Improve NonZero on CUDA/ROCM #10307 Merged

Improve NonZero on CUDA/ROCM #10307

onnxruntime
Improve NonZero on CUDA/ROCM
#10307

Merged