Skip to content

Commit 6a51e76

Browse files
authored
Modify condition for scale definition in sdpa_micro.cl
1 parent e01fa29 commit 6a51e76

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/plugins/intel_gpu/src/graph/impls/ocl_v2/sdpa_micro.cl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -544,7 +544,7 @@ KERNEL(micro_sdpa)(OPTIONAL_SHAPE_INFO_ARG
544544
#define greater_than(offset_k, offset_q) (offset_k > offset_q)
545545
#endif
546546

547-
#if IS_PAGED_ATTENTION && IS_PREFILL == 1
547+
#if IS_PAGED_ATTENTION && HAS_SINK_INPUT
548548
#define scale(x) ((x)* scale)
549549
tile_elementwise(S_tile, scale);
550550
#endif

0 commit comments

Comments
 (0)