@@ -2758,16 +2758,16 @@ __STATIC_INLINE__ ggml_tensor* ggml_ext_lokr_forward(
27582758 bool is_conv,
27592759 WeightAdapter::ForwardParams::conv2d_params_t conv_params,
27602760 float scale) {
2761- GGML_ASSERT ((w1 != NULL || (w1a != NULL && w1b != NULL )));
2762- GGML_ASSERT ((w2 != NULL || (w2a != NULL && w2b != NULL )));
2761+ GGML_ASSERT ((w1 != nullptr || (w1a != nullptr && w1b != nullptr )));
2762+ GGML_ASSERT ((w2 != nullptr || (w2a != nullptr && w2b != nullptr )));
27632763
2764- int uq = (w1 != NULL ) ? (int )w1->ne [0 ] : (int )w1a->ne [0 ];
2765- int up = (w1 != NULL ) ? (int )w1->ne [1 ] : (int )w1b->ne [1 ];
2764+ int uq = (w1 != nullptr ) ? (int )w1->ne [0 ] : (int )w1a->ne [0 ];
2765+ int up = (w1 != nullptr ) ? (int )w1->ne [1 ] : (int )w1b->ne [1 ];
27662766
27672767 int q_actual = is_conv ? (int )h->ne [2 ] : (int )h->ne [0 ];
27682768 int vq = q_actual / uq;
27692769
2770- int vp = (w2 != NULL ) ? (is_conv ? (int )w2->ne [3 ] : (int )w2->ne [1 ])
2770+ int vp = (w2 != nullptr ) ? (is_conv ? (int )w2->ne [3 ] : (int )w2->ne [1 ])
27712771 : (int )w2a->ne [1 ];
27722772 GGML_ASSERT (q_actual == (uq * vq) && " Input dimension mismatch for LoKR split" );
27732773
@@ -2803,7 +2803,7 @@ __STATIC_INLINE__ ggml_tensor* ggml_ext_lokr_forward(
28032803#endif
28042804
28052805 ggml_tensor* h_split = ggml_reshape_3d (ctx, h, vq, uq * merge_batch_uq, batch / merge_batch_uq);
2806- if (w2 != NULL ) {
2806+ if (w2 != nullptr ) {
28072807 hb = ggml_mul_mat (ctx, w2, h_split);
28082808 } else {
28092809 hb = ggml_mul_mat (ctx, w2b, ggml_mul_mat (ctx, w2a, h_split));
@@ -2816,7 +2816,7 @@ __STATIC_INLINE__ ggml_tensor* ggml_ext_lokr_forward(
28162816 hb_t = ggml_reshape_3d (ctx, hb_t , uq, vp * merge_batch_vp, batch / merge_batch_vp);
28172817
28182818 ggml_tensor* hc_t ;
2819- if (w1 != NULL ) {
2819+ if (w1 != nullptr ) {
28202820 hc_t = ggml_mul_mat (ctx, w1, hb_t );
28212821 } else {
28222822 hc_t = ggml_mul_mat (ctx, w1b, ggml_mul_mat (ctx, w1a, hb_t ));
@@ -2834,7 +2834,7 @@ __STATIC_INLINE__ ggml_tensor* ggml_ext_lokr_forward(
28342834 // 1. Reshape input: [W, H, vq*uq, batch] -> [W, H, vq, uq * batch]
28352835 ggml_tensor* h_split = ggml_reshape_4d (ctx, h, h->ne [0 ], h->ne [1 ], vq, uq * batch);
28362836
2837- if (w2 != NULL ) {
2837+ if (w2 != nullptr ) {
28382838 hb = ggml_ext_conv_2d (ctx, h_split, w2, nullptr ,
28392839 conv_params.s0 ,
28402840 conv_params.s1 ,
@@ -2902,7 +2902,7 @@ __STATIC_INLINE__ ggml_tensor* ggml_ext_lokr_forward(
29022902 ggml_tensor* hb_merged = ggml_reshape_2d (ctx, hb, w_out * h_out * vp, uq * batch);
29032903 ggml_tensor* hc_t ;
29042904 ggml_tensor* hb_merged_t = ggml_cont (ctx, ggml_transpose (ctx, hb_merged));
2905- if (w1 != NULL ) {
2905+ if (w1 != nullptr ) {
29062906 // Would be great to be able to transpose w1 instead to avoid transposing both hb and hc
29072907 hc_t = ggml_mul_mat (ctx, w1, hb_merged_t );
29082908 } else {
0 commit comments