EngineX-Ascend/enginex-ascend-910-vllm
11
0
Fork 1
Code Issues Pull Requests Actions Projects Releases Wiki Activity

init v0.23.0

Browse Source 
Signed-off-by: Sun Ruoxi <sunruoxi@4paradigm.com>
This commit is contained in:
Sun Ruoxi
2026-08-27 15:11:51 +08:00
parent b582a8e7d1
commit 7f8a1b1f7a
2849 changed files with 712887 additions and 22001 deletions
Download Patch File Download Diff File Expand all files Collapse all files

821
csrc/moe/scatter_nd_update_v2/op_host/config/ascend910_93/scatter_nd_update_v2_binary.json Normal file
View File

@@ -0,0 +1,821 @@
{
"op_type": "ScatterNdUpdateV2",
"op_list": [
{
"bin_filename": "ScatterNdUpdateV2_float16_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_float16_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_float32_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_float32_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bfloat16_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bfloat16_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int8_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int8_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int16_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int16_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int32_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int32_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int64_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int64_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bool_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bool_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
}
]
}

13
csrc/moe/scatter_nd_update_v2/op_host/config/ascend910_93/scatter_nd_update_v2_simplified_key.ini Normal file
View File

@@ -0,0 +1,13 @@
; 该文件主要影响 opc 工具 编译二进制kernel时， --simplified_key_mode 选项中填写的值，格式如下所示：
; [某算子]
; default=xx
; ascendxx=xx
; 其中，default为默认mode，ascnedxx为可选mode，如果不同芯片有差异化要求时，需要配置；
; 1)如果没有配置：非ascendC算子继续按空处理，即opc编译命令中不添加 --simplified_key_mode 选项，AscendC算子按照 simplified_key_mode=0 处理
; 2)如果仅有default配置：各个版本按default配置
; 3)如果仅有某些平台的配置，没有default配置：对应平台的按照配置的值传递，非对应平台的：非AscendC算子继续按空处理，AscendC算子按照 simplified_key_mode=0 处理
; 4)如果default配置和平台配置都有：对应平台的使用平台的配置，非对应的平台的以default值配置。
; 5)对于自定义simplified key的情况，需要在binary_simplified_key_mode.ini 文件中显式配置为None，不传入 --simplified_key_mode 选项，由opc工具和FE框架自行判断使用何种模式
; 6)是否是AscendC算子，由 ops/built-in/tbe/op_info_cfg/parser/ascendc_config.json 中配置的算子名字和对于的平台决定
[ScatterNdUpdateV2]
default=0

821
csrc/moe/scatter_nd_update_v2/op_host/config/ascend910b/scatter_nd_update_v2_binary.json Normal file
View File

@@ -0,0 +1,821 @@
{
"op_type": "ScatterNdUpdateV2",
"op_list": [
{
"bin_filename": "ScatterNdUpdateV2_float16_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_float16_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_float32_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_float32_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "float32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bfloat16_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bfloat16_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bfloat16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int8_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int8_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int8",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int16_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int16_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int16",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int32_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int32_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int64_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_int64_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bool_int32",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int32",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
},
{
"bin_filename": "ScatterNdUpdateV2_bool_int64",
"inputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "indices",
"index": 1,
"dtype": "int64",
"format": "ND",
"paramType": "required",
"shape": [-2]
},
{
"name": "updates",
"index": 2,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"outputs": [
{
"name": "var",
"index": 0,
"dtype": "bool",
"format": "ND",
"paramType": "required",
"shape": [-2]
}
],
"attrs": [
{
"name": "strides",
"dtype": "list_int",
"value": null
},
{
"name": "use_locking",
"dtype": "bool",
"value": null
}
]
}
]
}

13
csrc/moe/scatter_nd_update_v2/op_host/config/ascend910b/scatter_nd_update_v2_simplified_key.ini Normal file
View File

@@ -0,0 +1,13 @@
; 该文件主要影响 opc 工具 编译二进制kernel时， --simplified_key_mode 选项中填写的值，格式如下所示：
; [某算子]
; default=xx
; ascendxx=xx
; 其中，default为默认mode，ascnedxx为可选mode，如果不同芯片有差异化要求时，需要配置；
; 1)如果没有配置：非ascendC算子继续按空处理，即opc编译命令中不添加 --simplified_key_mode 选项，AscendC算子按照 simplified_key_mode=0 处理
; 2)如果仅有default配置：各个版本按default配置
; 3)如果仅有某些平台的配置，没有default配置：对应平台的按照配置的值传递，非对应平台的：非AscendC算子继续按空处理，AscendC算子按照 simplified_key_mode=0 处理
; 4)如果default配置和平台配置都有：对应平台的使用平台的配置，非对应的平台的以default值配置。
; 5)对于自定义simplified key的情况，需要在binary_simplified_key_mode.ini 文件中显式配置为None，不传入 --simplified_key_mode 选项，由opc工具和FE框架自行判断使用何种模式
; 6)是否是AscendC算子，由 ops/built-in/tbe/op_info_cfg/parser/ascendc_config.json 中配置的算子名字和对于的平台决定
[ScatterNdUpdateV2]
default=0