CANN: Support more ops (#12841)

* [CANN]Support Opt LOG && MEAN && PAD_REFLECT_1D * [CANN]Support COUNT_EQUAL && STEP && SGN * [CANN]codestyle adjustment * [CANN]codestyle adjustment --------- Signed-off-by: noemotiovon <noemotiovon@gmail.com>
2025-04-10 08:51:52 +08:00
parent 11d07e1e69
commit fe5b78c896
4 changed files with 173 additions and 0 deletions
--- a/ggml/src/ggml-cann/acl_tensor.cpp
+++ b/ggml/src/ggml-cann/acl_tensor.cpp
@@ -41,6 +41,8 @@ aclDataType ggml_cann_type_mapping(ggml_type type) {
            return ACL_INT4;
        case GGML_TYPE_Q8_0:
            return ACL_INT8;
+        case GGML_TYPE_I64:
+            return ACL_INT64;
        default:
            return ACL_DT_UNDEFINED;
    }