Merge branch 'main' of github.com:ROCm/vllm

ROCm · Mar 26, 2024 · d76bfa2 · d76bfa2
2 parents 6e8c375 + f8f77d9
commit d76bfa2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/layernorm_kernels.cu b/csrc/layernorm_kernels.cu
@@ -71,7 +71,7 @@ struct _typeConvert<c10::Half> {
   __device__ static inline packed_hip_type convert(float2 x) { return __float22half2_rn(x); }
 };
 
-#if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 800
+#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 800
 // CUDA_ARCH < 800 does not have BF16 support
 template<>
 struct _typeConvert<c10::BFloat16> {