onnxruntime
ResizeGrad CUDA/ROCM kernel implementation
#17772
Merged

ResizeGrad CUDA/ROCM kernel implementation #17772

baijumeswani merged 11 commits into main from baijumeswani/resize-grad
baijumeswani
baijumeswani ResizeGrad CUDA kernel implementation
5b7d4df9
baijumeswani baijumeswani added training
baijumeswani baijumeswani requested a review from askhade askhade 2 years ago
baijumeswani baijumeswani requested a review from pengwa pengwa 2 years ago
baijumeswani baijumeswani requested a review from Lafi7e Lafi7e 2 years ago
baijumeswani Register rocm kernels and address other pipeline failures
c7720475
baijumeswani baijumeswani added release:1.16.2
pengwa
pengwa commented on 2023-10-18
pengwa pengwa changed the title ResizeGrad CUDA kernel implementation ResizeGrad CUDA/ROCM kernel implementation 2 years ago
baijumeswani Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
7978819a
baijumeswani Address pull request review comments
b1bc9a5c
baijumeswani Use cuda only
e2e1b88f
baijumeswani Address pipeline failure
9ba2021d
pengwa
pengwa dismissed these changes on 2023-10-19
baijumeswani typo
581f9cd5
baijumeswani Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
95930ef2
baijumeswani baijumeswani dismissed their stale review via 95930ef2 2 years ago
baijumeswani Enable rocm ep for gradient checker test
b3589c81
baijumeswani Fix rocm error later
048caeea
pengwa
pengwa approved these changes on 2023-10-20
baijumeswani Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
046449f6
baijumeswani baijumeswani merged a43c57f5 into main 2 years ago
baijumeswani baijumeswani deleted the baijumeswani/resize-grad branch 2 years ago
tianleiwu tianleiwu removed release:1.16.2

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone