We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0bc67dd commit 2f538b9Copy full SHA for 2f538b9
ggml-cuda/common.cuh
@@ -306,6 +306,13 @@ static __device__ __forceinline__ half2 warp_reduce_max(half2 x) {
306
#endif // !(defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)) && __CUDA_ARCH__ >= CC_PASCAL && CUDART_VERSION >= CUDART_HMAX
307
}
308
309
+#if CUDART_VERSION < 12000
310
+static __device__ __forceinline__ uint __hgt2_mask(const half2 a, const half2 b) {
311
+ const uint mask_low = 0x0000FFFF * ( __low2half(a) > __low2half(b));
312
+ const uint mask_high = 0xFFFF0000 * (__high2half(a) > __high2half(b));
313
+ return mask_low | mask_high;
314
+}
315
+#endif // CUDART_VERSION < 12000
316
317
#if defined(GGML_USE_HIPBLAS)
318
#define __CUDA_ARCH__ 1300
0 commit comments