onnxruntime
Added Trilu CUDA kernel.
#9633
Merged
Overview
Commits
23
Changes
View On GitHub

Added Trilu CUDA kernel. #9633

satyajandhyala merged 23 commits into master from sajandhy/add_trilu_cuda_kernel
satyajandhyala
satyajandhyala satyajandhyala added training
satyajandhyala satyajandhyala added component:operator
satyajandhyala satyajandhyala added component:ortmodule
satyajandhyala satyajandhyala force pushed from e5d21c57 to 79ea9a49 4 years ago
satyajandhyala satyajandhyala force pushed from 7b5cb639 to 1f173c70 4 years ago
SherlockNoMad
SherlockNoMad commented on 2021-11-03
SherlockNoMad
SherlockNoMad commented on 2021-11-03
SherlockNoMad
SherlockNoMad commented on 2021-11-03
SherlockNoMad
SherlockNoMad commented on 2021-11-03
SherlockNoMad
SherlockNoMad commented on 2021-11-03
SherlockNoMad
satyajandhyala satyajandhyala force pushed from 32651409 to 70f60aea 4 years ago
satyajandhyala satyajandhyala force pushed from 70f60aea to b7a26df7 4 years ago
satyajandhyala satyajandhyala changed the title [WIP] Added Trilu CUDA kernel. Added Trilu CUDA kernel. 4 years ago
satyajandhyala satyajandhyala marked this pull request as ready for review 4 years ago
satyajandhyala satyajandhyala force pushed from b7a26df7 to 08c13559 4 years ago
satyajandhyala satyajandhyala force pushed from 08c13559 to 8e76c849 4 years ago
ytaous
satyajandhyala
satyajandhyala satyajandhyala requested a review from baijumeswani baijumeswani 4 years ago
satyajandhyala satyajandhyala requested a review from liqunfu liqunfu 4 years ago
satyajandhyala satyajandhyala requested a review from thiagocrepaldi thiagocrepaldi 4 years ago
satyajandhyala satyajandhyala requested a review from tlh20 tlh20 4 years ago
satyajandhyala satyajandhyala requested a review from xadupre xadupre 4 years ago
satyajandhyala satyajandhyala force pushed from f1a6fefb to 3514bc2b 4 years ago
satyajandhyala satyajandhyala force pushed from 3514bc2b to 2dc63ad3 4 years ago
satyajandhyala Added Trilu CUDA kernel.
a87277fb
satyajandhyala Fixed domain.
81decb6e
satyajandhyala Trilu is not a contrib op on CUDA.
446b71b9
satyajandhyala Handle Trilu optional second input k as a CPU tensor in CUDA kernel.
f2474894
satyajandhyala Revert formatting only changes.
e719614b
satyajandhyala Update trilu.cc
3e291f08
satyajandhyala Update trilu_impl.h
e1325786
satyajandhyala Revert "Trilu is not a contrib op on CUDA."
6294a95c
satyajandhyala Changed Trilu version 1 to a CUDA contib op.
39b84dde
satyajandhyala Formatting changes.
b8246eef
satyajandhyala Trilu should support multi-dim tensor input.
4cdf276d
satyajandhyala Fixed windows build.
4b806f42
satyajandhyala Updated OperatorKernels documentation.
33e87011
satyajandhyala Miscellanious Changes.
baed3691
satyajandhyala Fixed Windows build error.
b3f96fd3
satyajandhyala Added TriluGrad.
66030f07
satyajandhyala Added a training testcase for Trilu.
836f385e
satyajandhyala satyajandhyala force pushed from 2dc63ad3 to 836f385e 4 years ago
SherlockNoMad
SherlockNoMad commented on 2021-11-05
SherlockNoMad
SherlockNoMad commented on 2021-11-05
SherlockNoMad
SherlockNoMad commented on 2021-11-05
SherlockNoMad
SherlockNoMad commented on 2021-11-05
SherlockNoMad
SherlockNoMad commented on 2021-11-05
SherlockNoMad
SherlockNoMad commented on 2021-11-05
satyajandhyala Added more test data scenarios
46280728
satyajandhyala Changed the logic for a possible improvement.
e788e3e2
satyajandhyala Added optional input and attribute to the Trilu gradient builder
07bf960a
satyajandhyala Added Trilu gradient checker test.
dc986f19
satyajandhyala Added test parameters.
634903ef
satyajandhyala Changed variable type to auto
ba1b57cc
satyajandhyala satyajandhyala requested a review from SherlockNoMad SherlockNoMad 4 years ago
SherlockNoMad
SherlockNoMad approved these changes on 2021-11-09
satyajandhyala satyajandhyala merged 229c9a4e into master 4 years ago
satyajandhyala satyajandhyala deleted the sajandhy/add_trilu_cuda_kernel branch 4 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
training
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum