PR #3688 Refine GatherND CPU/CUDA Kernels & Add UTs

Refine GatherND CPU/CUDA Kernels & Add UTs #3688

pengwa merged 12 commits into master from pengwa/refacor_gather_nd

pengwa requested a review 5 years ago

pengwa changed the base branch from master to ort_training 5 years ago

pengwa requested a review from

AlexanderSher 5 years ago

pengwa requested a review from

SherlockNoMad 5 years ago

pengwa requested a review from

weixingzhang 5 years ago

pengwa removed review request from

AlexanderSher 5 years ago

pengwa changed the title ~~Refine GatherND CPU Kernel & Add UTs~~ Refine GatherND CPU/CUDA Kernels & Add UTs 5 years ago

pengwa added training

edgchen1 commented on 2020-04-24

SherlockNoMad commented on 2020-04-27

SherlockNoMad dismissed these changes on 2020-04-27

pengwa dismissed their stale review via f155aa43 5 years ago

pengwa force pushed from 34dec2dd to ef1081e8 5 years ago

pengwa changed the base branch from ort_training to master 5 years ago

Refactor GatherND CPU Kernel (Renaming & Simplify)

5ba22fa8

Add batch_dim=1 or 2, negative slices tests

1afadd83

Rename gather_nd_gard_impl.cu

d70dcabd

Use dispatcher to refactor CUDA GatherND/GatherNDGrad

266cf6d3

Fix builds

2daabcc8

Change GatherNDBase::CommonComputeKernel --> GatherNDBase::PrepareCom…

f07f3154

Use HasCudaEnvironment instead of __CUDA_ARCH__ for some double type …

5e90eb5a

Fix build error

9d53b76c

Resolve Comments

1000d418

pengwa force pushed from ef1081e8 to 1000d418 5 years ago

Fix build error

93a71456

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

730ce7c4

Fix build

98ccf21e

SherlockNoMad approved these changes on 2020-04-29

pengwa merged 0531accc into master 5 years ago

pengwa deleted the pengwa/refacor_gather_nd branch 5 years ago

Reviewers

SherlockNoMad

edgchen1

weixingzhang

Assignees

No one assigned

Labels

training

Milestone

No milestone

onnxruntime Refine GatherND CPU/CUDA Kernels & Add UTs #3688 Merged

Refine GatherND CPU/CUDA Kernels & Add UTs #3688

onnxruntime
Refine GatherND CPU/CUDA Kernels & Add UTs
#3688

Merged