Public
  1. Public

onnxruntime

Public

Commits

Vincent WangGitHub
7cac114e522
Vincent Wang authored and GitHub committed c07a3b869c5
Triton Codegen for ORTModule (#15831)

Fuse connected elementwise and reduce Ops to TritonOp and codegen triton
code to run the kernel.

This PR is co-edited by @wejoncy and @er3x3