未验证 提交 b2c797ad 编写于 作者: J Jeng Bai-Cheng 提交者: GitHub

Bugfix, CUB regression in CUDA 12.2 (#55594)

Issue #55016
上级 da2395f3
......@@ -32,9 +32,11 @@ __device__ inline float rsqrt(const float& x) {
return rsqrtf(x);
}
__device__ inline kv_float operator+(const kv_float& a, const kv_float& b) {
namespace cub {
__host__ __device__ inline kv_float operator+(const kv_float& a, const kv_float& b) {
return kv_float(a.key + b.key, a.value + b.value);
}
}
// Half Operations
__device__ inline half2 __hadd2_with_fallback(const half2 a, const half2 b) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册