#
Y GLSL.std.450 main $ h � � � � � � � 0 @ �
GL_GOOGLE_cpp_style_line_directive GL_GOOGLE_include_directive main v_tile gl_WorkGroupID bh tid gl_LocalInvocationID v_idx lane_on " Pc " seq_len " k_dim " v_dim $ + qk_bh 5 v_bh = gb_bh B state_bh J out_bh U j c s f State f state h x t � j � k_buf � K � k � � delta � decay � G � g � � beta_t � Beta � beta � � v_t � V � v � � kv_mem � j � j � q_buf � Q � q � y_t j . Out . outp 0 @ j G G H " # H " # H " # G " G e H f # G f G h " G h ! G � H � H � # G � G � " G � ! G � H � H � # G � G � " G � ! G � H � H � # G � G � " G � ! G � H � H � # G � G � " G � ! G � H � H � # G � G � " G � ! G - H . H . # G . G 0 " G 0 ! G X !
;
+
+ ;
+ @ " # " ; # $ % + % & ' + % - + % 1 _ + ` a _ ` b a e _ f e g f ; g h q _ t _ � a ; � � � _ � � � � ; � � � _ + � + � + _ � � _ � � � � ; � � � _ � � � � ; � � � _ � � � � ; � � ; � � � _ � � � � ; � � - _ . - / . ; / 0 , X 6 � ; ; ; ; ; ; + ; 5 ; = ; B ; J ; U ; b c ; x ; � ; t � ; t � ; t � ; t � ; t � ; � ; � ; t ; ; @ A
= > A
= > A
= > = � = � > = ! A ' ( $ & = ) ( � * ! ) > * = , A ' . $ - = / . � 0 , / A ' 2 $ 1 = 3 2 � 4 0 3 > + 4 = 6 A ' 7 $ - = 8 7 � 9 6 8 A ' : $ & = ; : � < 9 ; > 5 < = > A ' ? $ - = @ ? � A > @ > = A = C A ' D $ 1 = E D � F C E A ' G $ & = H G � I F H > B I = K A ' L $ - = M L � N K M A ' O $ & = P O � Q N P > J Q = R � T � R S T � S > U � V � V � X Y � Z � Z = [ U A ' \ $ 1 = ] \ � ^ [ ] � ^ W X � W = d U = i B = j U A ' k $ & = l k � m j l � n i m = o � p n o A q r h - p = _ s r A t u c d > u s � Y � Y = v U � w v 1 > U w � V � X � T � T > x � y � y � { | � } � } = ~ x A ' $ - = � � � ~ � � � z { � z = � > � � � � � � � � � � � � � = � � A ' � $ 1 = � � � � � � � � � � � � = � � = � + = � x A ' � $ 1 = � � � � � � � � � � = � � � � � � A q � � - � = _ � � A � � � � > � � � � � � = � � � � � > � � � � � � � � � � > � � = � � � � � � � � � = � = = � x � � � � A q � � - � = _ � � _ � � > � � = � = = � x � � � � A q � � - � = _ � � > � � = � 5 = � x A ' � $ & = � � � � � � � � � � = � � � � � A q � � - � = _ � � > � � > � � > � � � � � � � � � � � � = � � A ' � $ 1 = � � � � � � � � � � � � = � � = _ � � A t � c � = _ � � � _ � � � A t � c � > � � = � � A t � c � = _ � � = � � A � � � � = _ � � = _ � � _ � 2 � � � > � � � � � � = � � � � � 1 > � � � � � � = _ � � = _ � � � _ � � � = _ � � � _ � � � > � � � � � � = � > � � � � � � � � � � � � � = � � A ' � $ 1 = � � � � � � � � � � � � = � � = � + = � x A ' $ 1 = � � � � = � � A q � - = _ A � � � > � � � � = � �
> �
� � � � � � � � = �
�
� > � > � � � � � = A ' $ 1 = � � � = = A � � = _ = _ � = A t c = _ _ ! 2 A t " c > " ! = # A t $ c # = _ % $ = & A � ' � & = _ ( ' = _ ) _ * 2 % ( ) > * � � = + � , + 1 > , � � = 1 J = 2 x A ' 3 $ & = 4 3 � 5 2 4 � 6 1 5 = 7 � 8 6 7 = _ 9 A q : 0 - 8 > : 9 �
�
� � � � � | � | = ; x � < ; 1 > x < � y � { = = � ? � = > ? � > > @ � A � A � C D � E � E = F @ A ' G $ 1 = H G � I F H � I B C � B = J B = K @ A ' L $ & = M L � N K M � O J N = P � Q O P = R @ A t S c R = _ T S A q U h - Q > U T � D � D = V @ � W V 1 > @ W � A � C � ? � ? � 8