Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

ConvTransposeGrad CUDA Kernel #17201

Merged
merged 6 commits into from
Aug 24, 2023
Merged

ConvTransposeGrad CUDA Kernel #17201

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
e158626
ConvTransposeGrad CUDA Kernel
baijumeswani Aug 17, 2023
7cd4bbc
Reorder includes
baijumeswani Aug 17, 2023
ba00b10
Address pipeline error
baijumeswani Aug 18, 2023
fba872e
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
baijumeswani Aug 18, 2023
9028a8c
Avoid hipifying conv_transpose_grad
baijumeswani Aug 18, 2023
61c9c07
Disable ortmodule conv transpose grad tests for rocm pipeline
baijumeswani Aug 18, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Address pipeline error
  • Loading branch information
@baijumeswani
baijumeswani committed Aug 18, 2023
commit ba00b1048c12ab286f79041333f3f74db45fefc4
6 changes: 3 additions & 3 deletions orttraining/orttraining/training_ops/cuda/nn/conv_shared.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -268,8 +268,8 @@ Status AlgoIterator<T_Perf>::TryAll(const CUDAExecutionProvider* provider, const
return Status::OK();
}

template struct AlgoIterator<T_BwdDataPerf>;
template struct AlgoIterator<T_BwdFilterPerf>;
template struct AlgoIterator<T_FwdPerf>;
template class AlgoIterator<T_BwdDataPerf>;
template class AlgoIterator<T_BwdFilterPerf>;
template class AlgoIterator<T_FwdPerf>;

} // namespace onnxruntime::cuda