"vscode:/vscode.git/clone" did not exist on "df15703b422644c7cbbb9779af9e8f1fd639eefe"
Unverified Commit bf8b022e authored by Mieszko Dziadowiec's avatar Mieszko Dziadowiec Committed by GitHub
Browse files

[Intel][Triton] Support `round_int8` for Intel backend (#38825)


Signed-off-by: default avatarMieszko Dziadowiec <mdziadowiec@habana.ai>
Signed-off-by: default avatarLucas Wilkinson <lwilkins@redhat.com>
Signed-off-by: default avatarStefano Castagnetta <scastagnetta@nvidia.com>
Co-authored-by: default avatarLucas Wilkinson <LucasWilkinson@users.noreply.github.com>
Co-authored-by: default avatarStefano Castagnetta <scastagnetta@nvidia.com>
Co-authored-by: default avatarClaude <noreply@anthropic.com>
Co-authored-by: default avatarKunshang Ji <kunshang.ji@intel.com>
parent 40ee64c0
......@@ -88,6 +88,13 @@ if current_platform.is_rocm():
def round_int8(x):
return tl.extra.hip.libdevice.round(x).to(tl.int8)
elif current_platform.is_xpu():
@triton.jit
def round_int8(x):
return tl.extra.intel.libdevice.round(x).to(tl.int8)
else:
@triton.jit
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment