TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

✨[Feature] Add reduction targets in Autocast

Open zewenli98 opened this issue 1 month ago • 2 comments

Need to add reduction targets in Autocast DepthOfReductionRule

zewenli98 avatar Dec 05 '25 19:12 zewenli98

@zewenli98 do you have specific ones we need to add?

narendasan avatar Dec 07 '25 21:12 narendasan

Will go over the current reduction ops and see what should be added. btw the definition of reduction op: https://en.wikipedia.org/wiki/Reduction_operator

zewenli98 avatar Dec 11 '25 01:12 zewenli98