Skip to content

Commit

Permalink
add hipblaslt support
Browse files Browse the repository at this point in the history
  • Loading branch information
ScXfjiang committed Dec 4, 2024
1 parent 5b1bf93 commit 36c4c0b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion xla/stream_executor/device_description.h
Original file line number Diff line number Diff line change
Expand Up @@ -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(); }

Expand Down

0 comments on commit 36c4c0b

Please sign in to comment.