Question about kernel implementation Compute() method #20992
Unanswered
dgiantsidi
asked this question in
EP Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi there,
What is the difference between contrib_ops and core/providers regarding the kernels' Compute method? What kind of kernels are implemented in the core/providers directories?
I want to print all related info about a OpKernelContext object (kernel name, input arguments, output after computation) at the Compute method and all management-related events (cudaMalloc() and other API).
Similarly, how can I print all related info about the graph that is constructed after optimizations and right before its scheduling?
Can anyone provide some guidance?
Thanks,
Dimitra
Beta Was this translation helpful? Give feedback.
All reactions