[ROCm][V0][Attention] Revert to the previous FA triton kernel (vllm-project#18226)

Signed-off-by: Gregory Shtrasberg <[email protected]>

Author: gshtras

vllm/attention/backends/rocm_flash_attn.py

@@ -770,8 +770,9 @@ def forward(
                                       and layer._v_scale and layer._prob_scale
                                       and self.kv_cache_dtype == "fp8")
                     full_scales = (
-                        layer._q_scale, layer._k_scale, layer._v_scale,
-                        layer._prob_scale) if use_fp8_scales else None
+                        layer._q_scale.item(), layer._k_scale.item(),
+                        layer._v_scale.item(),
+                        layer._prob_scale.item()) if use_fp8_scales else None
                     self.triton_attn_func(
                         query,
                         key,