musa: support flash attention (#9)

Signed-off-by: Xiaodong Ye <[email protected]>

Author: yeahdongcn

discover/types.go

@@ -184,7 +184,8 @@ func (l GpuInfoList) FlashAttentionSupported() bool {
 		supportsFA := gpu.Library == "cpu" ||
 			gpu.Library == "metal" ||
 			(gpu.Library == "cuda" && gpu.DriverMajor >= 7) ||
-			gpu.Library == "rocm"
+			gpu.Library == "rocm" ||
+			gpu.Library == "musa"
 
 		if !supportsFA {
 			return false