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>
This commit is contained in:
parent
11d07e1e69
commit
fe5b78c896
4 changed files with 173 additions and 0 deletions
|
@ -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;
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue