Triton current scaling: avoid casting amax input#458

matthiasdiener wants to merge 4 commits intodevROCm/TransformerEngine:devfrom

mdiener/triton-castROCm/TransformerEngine:mdiener/triton-castCopy head branch name to clipboard