From 36c4c0b2227aac3d06b9985fbb8accf0aa324fb2 Mon Sep 17 00:00:00 2001 From: scxfjiang Date: Wed, 4 Dec 2024 01:37:21 -0800 Subject: [PATCH] add hipblaslt support --- xla/stream_executor/device_description.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/xla/stream_executor/device_description.h b/xla/stream_executor/device_description.h index ae8798a5d2c4f8..f22a591995a610 100644 --- a/xla/stream_executor/device_description.h +++ b/xla/stream_executor/device_description.h @@ -231,7 +231,7 @@ class RocmComputeCapability { return gfx_version() != "gfx900" && gfx_version() != "gfx906"; } - bool has_hipblaslt() const { return gfx9_mi200_or_later(); } + bool has_hipblaslt() const { return gfx9_mi200_or_later() || gfx12_rx8600() || gfx12_rx8800(); } bool has_fp8_support() const { return gfx9_mi300() || gfx12_rx8600() || gfx12_rx8800(); }