Skip to content

GitLab

Merged
Created by hanjj@hanjj1Developer

fix(cub):fix pytorch bug 114942

编译pytorch 会添加宏-D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__；这些宏影响fp16和bf16一些重载函数

@中科海光--DTK软件质量部