下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

TensorPlacementUtils

函数功能

提供一组函数,判断TensorPlacement的位置。

函数原型

class TensorPlacementUtils {
 public:
  // 判断Tensor是否位于Device上的内存
  static bool IsOnDevice(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Host上
  static bool IsOnHost(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Host上,且数据紧跟在结构体后面
  static bool IsOnHostFollowing(TensorPlacement placement) {
   ...
  }
  // 判断Tensor是否位于Host上,且数据不紧跟在结构体后面
  static bool IsOnHostNotFollowing(TensorPlacement placement) {
   ...
  }
  // 判断Tensor是否位于Device上的内存
  static bool IsOnDeviceHbm(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Device上的P2p内存
  static bool IsOnDeviceP2p(TensorPlacement placement) {
    ...
  }
};

参数说明

参数

输入/输出

说明

placement

输入

需要进行判断的TensorPlacement枚举。

返回值说明

true表示是

false表示不是

约束说明

无。

调用示例

TensorData tensor_data;
tensor_data.SetPlacement(TensorPlacement::kOnHost);
auto on_host = TensorPlacementUtils::IsOnHost(tensor_data.GetPlacement()); // on_host is true
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词