昇腾社区首页
中文
注册

class CompileSpec

说明

用于配置compile接口所需要的参数。

定义

GraphInputs graphInputs;
PrecisionPolicy precision_policy = PrecisionPolicy::FP16;
std::string soc_version = "Ascend310P3";
size_t minBlockSize = 1;
size_t optimizationLevel = 0;
bool requireFullCompilation = false;
bool truncateLongAndDouble = true;
bool allowTensorReplaceInt = false;
std::vector<std::string> torchExecutedOps;
std::vector<std::string> torchExecutedModules;