CANN opbase 数据类型工具IsIntegralType 整数类型判定接口解析【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase本指南围绕 CANN opbase 算子库中op命名空间下的IsIntegralType接口展开系统讲解其功能语义、函数原型、判定范围与源码实现并结合头文件与单元测试给出算子开发中的典型使用场景。读完本文你将掌握如何在算子实现中快速判断ge::DataType是否为整数类型并了解它与IsFloatingType、IsComplexType、CanCast等数据类型工具函数的协作方式。接口定位opdev 数据类型工具族中的一员在 CANN opbase 算子库中算子开发者需要频繁处理ge::DataType枚举类型例如在 Shape 推导、tiling 计算、参数校验等环节判断输入张量的数据类型。为统一此类判断逻辑opbase 在 data_type_utils 接口总览 下提供了一组以Is、To、Check等动词命名的工具函数包括类型分类判断IsBasicType、IsNumberType、IsRealNumberType、IsComplexType、IsFloatingType、IsIntegralType类型转换ToDataType、ToOpDataType、ToAclDataType、ToString类型运算与查询TypeSize、CanCast、PromoteType、CheckTypeIsIntegralType是其中用于整数类型判定的核心函数也是CanCast、PromoteType等复合逻辑的底层依赖之一。功能说明与判定范围IsIntegralType用于判断输入的数据类型是否为整数类型。根据接口文档被判定为整数类型的包括以下 8 种整数类型对应的 ge::DataType 枚举值位宽Int8DT_INT88 位有符号Int16DT_INT1616 位有符号Int32DT_INT3232 位有符号Int64DT_INT6464 位有符号Uint8DT_UINT88 位无符号Uint16DT_UINT1616 位无符号Uint32DT_UINT3232 位无符号Uint64DT_UINT6464 位无符号值得注意的是boolDT_BOOL不在单参数版本的整数判定范围内。opbase 为 bool 单独提供了双参数重载版本详见后文使用时需要显式传入开关。函数原型与参数说明单参数版本的函数原型如下bool IsIntegralType(const ge::DataType type)参数输入/输出说明type输入输入的数据类型类型为ge::DataType枚举。type参数用于传入待判定的数据类型实际取值来自ge::DataType枚举例如ge::DataType::DT_INT32。接口以值传递方式接收参数不会修改调用方数据也不依赖外部状态因此可以在热路径中安全调用。返回值说明若type属于上述 8 种整数类型之一返回true否则返回false包括浮点、复数、bool、字符串等非整数类型。源码实现解析IsIntegralType定义在 include/nnopbase/opdev/data_type_utils.h 中实现为头文件内联函数通过逐一与 8 个整数类型枚举值比较得出结果inline bool IsIntegralType(const ge::DataType type) { return (type ge::DataType::DT_INT8 || type ge::DataType::DT_INT16 || type ge::DataType::DT_INT32 || type ge::DataType::DT_INT64 || type ge::DataType::DT_UINT8 || type ge::DataType::DT_UINT16 || type ge::DataType::DT_UINT32 || type ge::DataType::DT_UINT64); }从源码结构可以归纳出三个实现要点全枚举显式比较实现没有使用CheckType的容器查找方式而是用 8 个||短路比较直接判定。由于ge::DataType是枚举类型编译期即可完成常量折叠运行时开销极低适合在算子的高频校验路径中反复调用。inline 内联设计函数体声明为inline定义于头文件与IsFloatingType、IsComplexType、TypeSize等接口同文件并列保证零额外调用开销也便于算子侧直接包含头文件使用。DT_MAX 之外的防御边界同文件中的 TypeSize 等函数会先检查枚举值是否落在[DT_FLOAT, DT_MAX)合法区间内而IsIntegralType对任何未列出的枚举值一律返回false语义上等价于未知类型不属于整数类型。同一头文件在文件顶部还定义了若干短名常量如i1 DT_INT8、i4 DT_INT32、u1 DT_UINT8等用于kPromoteTypesLookup类型提升表的紧凑书写这也从侧面说明 8 种整数类型在类型运算体系中的基础地位。含 bool 的重载版本IsIntegralType含 boolopbase 额外提供了带include_bool参数的重载版本专门用于需要把 bool 视为整数的场景其文档见 IsIntegralType含boolbool IsIntegralType(const ge::DataType type, const bool include_bool)参数输入/输出说明type输入输入的数据类型。include_bool输入是否将 bool 视为整数类型。源码实现如下include/nnopbase/opdev/data_type_utils.hinline bool IsIntegralType(const ge::DataType type, const bool include_bool) { bool is_integral IsIntegralType(type); return include_bool ? (is_integral || (type ge::DataType::DT_BOOL)) : is_integral; }include_bool为true时DT_BOOL也被视为整数类型include_bool为false时行为与单参数版本完全一致。这一设计体现了 opbase 对bool 是否算整数这一语义分歧的显式处理默认不将 bool 归入整数需要时由调用方显式声明避免在算子间产生隐式约定不一致。调用示例基础用法非整数类型提前返回以下示例来自接口文档用于在算子入口校验数据类型// 校验dtype是否为整数类型不是则提前返回 void Func(const ge::DataType type) { if (!IsIntegralType(type)) { return; } // 后续执行算子计算逻辑 }含 bool 的用法// 校验dtype是否为整数类型含bool不是则提前返回 void Func(const ge::DataType type) { if (!IsIntegralType(type, true)) { return; } // 后续执行算子计算逻辑 }与错误码结合的参数校验在实际算子实现中通常配合日志与错误码使用例如仅当类型为整数时才执行后续逻辑否则记录错误并返回#include opdev/data_type_utils.h aclnnStatus Func(const ge::DataType type) { if (!op::IsIntegralType(type)) { // 记录错误日志并返回失败码例如 ACLNN_INVALID_PARAM return ACLNN_INVALID_PARAM; } // 执行整数类型专属的算子计算逻辑 return ACLNN_SUCCESS; }在算子开发中的典型应用场景IsIntegralType在 opbase 的类型体系中承担整数性判定这一基础职责从同头文件的源码可以确认它与以下函数存在直接协作关系CanCast的类型安全校验include/nnopbase/opdev/data_type_utils.hCanCast在判断浮点类型能否强转为整数类型时调用IsIntegralType(to, false)明确禁止浮点到整数的隐式转换防止精度丢失inline bool CanCast(const ge::DataType from, const ge::DataType to) { if (IsComplexType(from) !IsComplexType(to)) { return false; } if (IsFloatingType(from) IsIntegralType(to, false)) { return false; } if (from ! ge::DataType::DT_BOOL to ge::DataType::DT_BOOL) { return false; } return true; }与IsFloatingType、IsComplexType互补同头文件中的IsFloatingType覆盖浮点/低精度浮点含DT_BF16、DT_FLOAT8_E4M3FN、DT_FLOAT4_E2M1等新型低精度类型IsComplexType覆盖三种复数类型。三者共同构成整数 / 浮点 / 复数的类型三分法算子侧可按此组合实现完整的类型分支。单元测试印证仓库的 test_data_type_utils.cpp 对同族接口CanCast、CheckType等进行了断言验证如EXPECT_TRUE(op::CanCast(op::DataType::DT_INT32, op::DataType::DT_FLOAT))从侧面印证了IsIntegralType作为底层判定的正确性直接影响上层转换语义。注意事项与版本提示头文件 include/nnopbase/opdev/data_type_utils.h 中定义的接口位于op命名空间是当前推荐使用的版本。仓库中还存在旧版头文件 include/nnopbase/opdev/op_common/data_type_utils.h其函数位于opcommon命名空间功能与新版基本一致但头文件内带有#warning编译告警提示该头文件预计于 2027 年 6 月弃用建议迁移到include/nnopbase/opdev下的新接口。单参数版本不把 bool 视为整数若业务语义需要将 bool 纳入整数范围请使用双参数重载并传入include_bool true。相关资源索引接口文档IsIntegralType、IsIntegralType含bool、data_type_utils 接口总览头文件include/nnopbase/opdev/data_type_utils.hop命名空间推荐、include/nnopbase/opdev/op_common/data_type_utils.hopcommon命名空间将弃用源文件src/nnopbase/common/utils/data_type_utils.cppToString、ToDataType、CalcShapeBytes等非内联实现测试用例tests/nnopbase/ut/composite_op/test_data_type_utils.cpp、tests/nnopbase/st/composite_op/test_data_type_utils.cpp【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
CANN opbase 数据类型工具：IsIntegralType 整数类型判定接口解析
CANN opbase 数据类型工具IsIntegralType 整数类型判定接口解析【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase本指南围绕 CANN opbase 算子库中op命名空间下的IsIntegralType接口展开系统讲解其功能语义、函数原型、判定范围与源码实现并结合头文件与单元测试给出算子开发中的典型使用场景。读完本文你将掌握如何在算子实现中快速判断ge::DataType是否为整数类型并了解它与IsFloatingType、IsComplexType、CanCast等数据类型工具函数的协作方式。接口定位opdev 数据类型工具族中的一员在 CANN opbase 算子库中算子开发者需要频繁处理ge::DataType枚举类型例如在 Shape 推导、tiling 计算、参数校验等环节判断输入张量的数据类型。为统一此类判断逻辑opbase 在 data_type_utils 接口总览 下提供了一组以Is、To、Check等动词命名的工具函数包括类型分类判断IsBasicType、IsNumberType、IsRealNumberType、IsComplexType、IsFloatingType、IsIntegralType类型转换ToDataType、ToOpDataType、ToAclDataType、ToString类型运算与查询TypeSize、CanCast、PromoteType、CheckTypeIsIntegralType是其中用于整数类型判定的核心函数也是CanCast、PromoteType等复合逻辑的底层依赖之一。功能说明与判定范围IsIntegralType用于判断输入的数据类型是否为整数类型。根据接口文档被判定为整数类型的包括以下 8 种整数类型对应的 ge::DataType 枚举值位宽Int8DT_INT88 位有符号Int16DT_INT1616 位有符号Int32DT_INT3232 位有符号Int64DT_INT6464 位有符号Uint8DT_UINT88 位无符号Uint16DT_UINT1616 位无符号Uint32DT_UINT3232 位无符号Uint64DT_UINT6464 位无符号值得注意的是boolDT_BOOL不在单参数版本的整数判定范围内。opbase 为 bool 单独提供了双参数重载版本详见后文使用时需要显式传入开关。函数原型与参数说明单参数版本的函数原型如下bool IsIntegralType(const ge::DataType type)参数输入/输出说明type输入输入的数据类型类型为ge::DataType枚举。type参数用于传入待判定的数据类型实际取值来自ge::DataType枚举例如ge::DataType::DT_INT32。接口以值传递方式接收参数不会修改调用方数据也不依赖外部状态因此可以在热路径中安全调用。返回值说明若type属于上述 8 种整数类型之一返回true否则返回false包括浮点、复数、bool、字符串等非整数类型。源码实现解析IsIntegralType定义在 include/nnopbase/opdev/data_type_utils.h 中实现为头文件内联函数通过逐一与 8 个整数类型枚举值比较得出结果inline bool IsIntegralType(const ge::DataType type) { return (type ge::DataType::DT_INT8 || type ge::DataType::DT_INT16 || type ge::DataType::DT_INT32 || type ge::DataType::DT_INT64 || type ge::DataType::DT_UINT8 || type ge::DataType::DT_UINT16 || type ge::DataType::DT_UINT32 || type ge::DataType::DT_UINT64); }从源码结构可以归纳出三个实现要点全枚举显式比较实现没有使用CheckType的容器查找方式而是用 8 个||短路比较直接判定。由于ge::DataType是枚举类型编译期即可完成常量折叠运行时开销极低适合在算子的高频校验路径中反复调用。inline 内联设计函数体声明为inline定义于头文件与IsFloatingType、IsComplexType、TypeSize等接口同文件并列保证零额外调用开销也便于算子侧直接包含头文件使用。DT_MAX 之外的防御边界同文件中的 TypeSize 等函数会先检查枚举值是否落在[DT_FLOAT, DT_MAX)合法区间内而IsIntegralType对任何未列出的枚举值一律返回false语义上等价于未知类型不属于整数类型。同一头文件在文件顶部还定义了若干短名常量如i1 DT_INT8、i4 DT_INT32、u1 DT_UINT8等用于kPromoteTypesLookup类型提升表的紧凑书写这也从侧面说明 8 种整数类型在类型运算体系中的基础地位。含 bool 的重载版本IsIntegralType含 boolopbase 额外提供了带include_bool参数的重载版本专门用于需要把 bool 视为整数的场景其文档见 IsIntegralType含boolbool IsIntegralType(const ge::DataType type, const bool include_bool)参数输入/输出说明type输入输入的数据类型。include_bool输入是否将 bool 视为整数类型。源码实现如下include/nnopbase/opdev/data_type_utils.hinline bool IsIntegralType(const ge::DataType type, const bool include_bool) { bool is_integral IsIntegralType(type); return include_bool ? (is_integral || (type ge::DataType::DT_BOOL)) : is_integral; }include_bool为true时DT_BOOL也被视为整数类型include_bool为false时行为与单参数版本完全一致。这一设计体现了 opbase 对bool 是否算整数这一语义分歧的显式处理默认不将 bool 归入整数需要时由调用方显式声明避免在算子间产生隐式约定不一致。调用示例基础用法非整数类型提前返回以下示例来自接口文档用于在算子入口校验数据类型// 校验dtype是否为整数类型不是则提前返回 void Func(const ge::DataType type) { if (!IsIntegralType(type)) { return; } // 后续执行算子计算逻辑 }含 bool 的用法// 校验dtype是否为整数类型含bool不是则提前返回 void Func(const ge::DataType type) { if (!IsIntegralType(type, true)) { return; } // 后续执行算子计算逻辑 }与错误码结合的参数校验在实际算子实现中通常配合日志与错误码使用例如仅当类型为整数时才执行后续逻辑否则记录错误并返回#include opdev/data_type_utils.h aclnnStatus Func(const ge::DataType type) { if (!op::IsIntegralType(type)) { // 记录错误日志并返回失败码例如 ACLNN_INVALID_PARAM return ACLNN_INVALID_PARAM; } // 执行整数类型专属的算子计算逻辑 return ACLNN_SUCCESS; }在算子开发中的典型应用场景IsIntegralType在 opbase 的类型体系中承担整数性判定这一基础职责从同头文件的源码可以确认它与以下函数存在直接协作关系CanCast的类型安全校验include/nnopbase/opdev/data_type_utils.hCanCast在判断浮点类型能否强转为整数类型时调用IsIntegralType(to, false)明确禁止浮点到整数的隐式转换防止精度丢失inline bool CanCast(const ge::DataType from, const ge::DataType to) { if (IsComplexType(from) !IsComplexType(to)) { return false; } if (IsFloatingType(from) IsIntegralType(to, false)) { return false; } if (from ! ge::DataType::DT_BOOL to ge::DataType::DT_BOOL) { return false; } return true; }与IsFloatingType、IsComplexType互补同头文件中的IsFloatingType覆盖浮点/低精度浮点含DT_BF16、DT_FLOAT8_E4M3FN、DT_FLOAT4_E2M1等新型低精度类型IsComplexType覆盖三种复数类型。三者共同构成整数 / 浮点 / 复数的类型三分法算子侧可按此组合实现完整的类型分支。单元测试印证仓库的 test_data_type_utils.cpp 对同族接口CanCast、CheckType等进行了断言验证如EXPECT_TRUE(op::CanCast(op::DataType::DT_INT32, op::DataType::DT_FLOAT))从侧面印证了IsIntegralType作为底层判定的正确性直接影响上层转换语义。注意事项与版本提示头文件 include/nnopbase/opdev/data_type_utils.h 中定义的接口位于op命名空间是当前推荐使用的版本。仓库中还存在旧版头文件 include/nnopbase/opdev/op_common/data_type_utils.h其函数位于opcommon命名空间功能与新版基本一致但头文件内带有#warning编译告警提示该头文件预计于 2027 年 6 月弃用建议迁移到include/nnopbase/opdev下的新接口。单参数版本不把 bool 视为整数若业务语义需要将 bool 纳入整数范围请使用双参数重载并传入include_bool true。相关资源索引接口文档IsIntegralType、IsIntegralType含bool、data_type_utils 接口总览头文件include/nnopbase/opdev/data_type_utils.hop命名空间推荐、include/nnopbase/opdev/op_common/data_type_utils.hopcommon命名空间将弃用源文件src/nnopbase/common/utils/data_type_utils.cppToString、ToDataType、CalcShapeBytes等非内联实现测试用例tests/nnopbase/ut/composite_op/test_data_type_utils.cpp、tests/nnopbase/st/composite_op/test_data_type_utils.cpp【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
