FlatAT

python3 flat_at_generate_model.py --cores <core_num> -d <dim> -c <code_num> -p <process_id> -t <npu_type>

<dim>：输入特征向量维度，默认值为“64”。

<code_num>：与输入特征作对比的底库特征数，默认值为“8192”。

<core_num>：昇腾AI处理器AI Core的个数，默认为“2”。不指定该值时，根据<npu_type>配置：当npu_type配置为310时，<core_num>配置为2；当npu_type配置为310P时，<core_num>配置为8。

<process_id>：批量生成算子多进程调度的进程ID，默认值为“0”，无需设置。

<npu_type>：硬件形态，当前<npu_type>支持Atlas 200/300/500 推理产品以及Atlas 推理系列产品，取值范围分别为：310、310P，默认为“310”。

--help | -h：查询帮助信息。

执行此命令，用户可以得到一组算子模型文件。

FlatAT算子主要用于在IVF场景，减少train和add的耗时。

• dim ∈ {64, 128, 256}
• code_num ∈ {1024, 2048, 4096, 8192, 16384, 32768}