ggml-zdnn: code cleanup

Signed-off-by: Aaron Teo <[email protected]>

Author: taronaeo

ggml/src/ggml-zdnn/ggml-zdnn.cpp

@@ -20,10 +20,14 @@ static void ggml_zdnn_compute_forward_mul_mat(
     const ggml_tensor * src1 = dst->src[1];  // inputs
 
     // TODO: implement support for quantized types
+    // we currently only support f32, f16, and bf16
     ggml_zdnn_mul_mat_f(ctx, src0, src1, dst);
 }
 
-static bool ggml_zdnn_compute_forward(ggml_backend_zdnn_context * ctx, ggml_tensor * dst) {
+static bool ggml_zdnn_compute_forward(
+    ggml_backend_zdnn_context * ctx,
+    ggml_tensor * dst) {
+
     switch (dst->op) {
         case GGML_OP_MUL_MAT:
             {

ggml/src/ggml-zdnn/utils.cpp

@@ -23,11 +23,11 @@ zdnn_data_types ggml_zdnn_type_mapping(ggml_type type) {
 }
 
 void ggml_zdnn_create_tensor(zdnn_tensor_desc  & pre_tfm_desc,
-                                    zdnn_tensor_desc  & tfm_desc,
-                                    zdnn_ztensor      & ztensor,
-                              const ggml_tensor       * src,
-                              const int64_t           * ne,
-                              const zdnn_data_layouts   layout) {
+                             zdnn_tensor_desc  & tfm_desc,
+                             zdnn_ztensor      & ztensor,
+                       const ggml_tensor       * src,
+                       const int64_t           * ne,
+                       const zdnn_data_layouts   layout) {
     zdnn_init_pre_transformed_desc(
         layout,
         ggml_zdnn_type_mapping(src->type),

ggml/src/ggml-zdnn/utils.hpp

@@ -5,12 +5,12 @@
 
 zdnn_data_types ggml_zdnn_type_mapping(ggml_type type);
 
-void ggml_zdnn_create_tensor(zdnn_tensor_desc  & pre_tfm_desc,
-                                    zdnn_tensor_desc  & tfm_desc,
-                                    zdnn_ztensor      & ztensor,
-                              const ggml_tensor       * src,
-                              const int64_t           * ne,
-                              const zdnn_data_layouts   layout);
+void ggml_zdnn_create_tensor(zdnn_tensor_desc & pre_tfm_desc,
+                             zdnn_tensor_desc & tfm_desc,
+                             zdnn_ztensor     & ztensor,
+                      const ggml_tensor       * src,
+                      const int64_t           * ne,
+                      const zdnn_data_layouts   layout);
 
 void ggml_zdnn_load_tensor(zdnn_ztensor & ztensor, void * buffer);