Star1569
1576
代码
代码
Issues182
Pull Requests416
Actions
项目讨论
Wiki
分析
Star1569
1576
已合并
stateless_random_normal_v2 support bf16 #1907
Ding_Jing创建于 3月26日
stateless_random_normal_v2 support bf16 #1907
已合并
Ding_Jing创建于 3月26日
从Ding_Jing/ops-mathadd_bf16_0326合入到cann/ops-mathmaster
共 5 个文件变更+9-9
Mrandom/stateless_random_normal_v2/docs/aclnnNormalFloatFloat.md+1-1文件内容审核中，请稍后刷新重试
Mrandom/stateless_random_normal_v2/docs/aclnnNormalFloatTensor.md+2-2文件内容审核中，请稍后刷新重试
Mrandom/stateless_random_normal_v2/docs/aclnnNormalTensorFloat.md+2-2文件内容审核中，请稍后刷新重试
Mrandom/stateless_random_normal_v2/docs/aclnnNormalTensorTensor.md+3-3文件内容审核中，请稍后刷新重试
Mrandom/stateless_random_normal_v2/op_api/aclnn_normal_out.cpp+1-1
@@ -34,7 +34,7 @@ extern "C" {
34static constexpr size_t MAX_DIM_LEN = 8;34static constexpr size_t MAX_DIM_LEN = 8;
35 35 
36static const std::initializer_list<op::DataType> DTYPE_SUPPORT_LIST = {36static const std::initializer_list<op::DataType> DTYPE_SUPPORT_LIST = {
37- op::DataType::DT_FLOAT16, op::DataType::DT_FLOAT, op::DataType::DT_DOUBLE};37+ op::DataType::DT_BF16, op::DataType::DT_FLOAT16, op::DataType::DT_FLOAT, op::DataType::DT_DOUBLE};
38 38 
39/* 查看TensorFloat的Dtype和Shape */39/* 查看TensorFloat的Dtype和Shape */
40static bool CheckTensorAndFloatDtype(const aclTensor* mean, const aclTensor* out)40static bool CheckTensorAndFloatDtype(const aclTensor* mean, const aclTensor* out)