From 6fc40bbfce7149c8dca80936934756f60a4c8c65 Mon Sep 17 00:00:00 2001 From: Juuso Korhonen <40278371+juuso-oskari@users.noreply.github.com> Date: Wed, 18 Dec 2024 13:08:38 +0000 Subject: [PATCH] fix --- python/perf-kernels/flash-attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/perf-kernels/flash-attention.py b/python/perf-kernels/flash-attention.py index dbd1b64b1f32..f013dc54ab8b 100644 --- a/python/perf-kernels/flash-attention.py +++ b/python/perf-kernels/flash-attention.py @@ -63,7 +63,7 @@ def set_varlen_params(self, cu_seqlens_q, cu_seqlens_k): def set_persistent(self, persistent): self.persistent = persistent - + def set_int8_params(self, q_descale, k_descale, v_descale, p_scale, p_descale): self.int8 = True self.q_descale = q_descale