fix: restore cuda_bf16.h include

This commit is contained in:
2026-05-28 05:11:39 +00:00
parent 8783a25deb
commit 1734d13f60

View File

@@ -30,14 +30,8 @@
#pragma once
#include <cuda_runtime.h>
#include <cstdint>
// __nv_bfloat16 is a built-in type on CUDA 13+
// cuda_bf16.h may have C++17 compatibility issues with some CUDA versions;
// only include it in device code where it's guaranteed to work.
#if defined(__CUDA_ARCH__)
#include <cuda_bf16.h>
#endif
#include <cstdint>
// CUTLASS C++ includes (CUDA device code only)
#if defined(__CUDA_ARCH__)