deepseek-ai / deepseek-ai/FlashMLA

编译错误,怎么解决?有人遇到么

Open
#17 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
C++
Stars
12.9k
Forks
1.2k
Avg merge
4h 20m
Merged PRs (30d)
2

Description

D:\GitHub\deepseek-ai\FlashMLA\csrc\flash_fwd_mla_kernel.h(522): warning #221-D: floating-point value does not fit in required floating-point type
max_lse = max_lse == -((float)(1e+300)) ? 0.0f : max_lse;
^
detected during:
instantiation of "void flash::flash_fwd_splitkv_mla_combine_kernel(Flash_fwd_mla_params) [with Element=cutlass::bfloat16_t, ElementAccum=float, index_t=int64_t, kHeadDimV=512, kMaxSplits=160]" at line 588
instantiation of "void run_flash_splitkv_fwd_mla(Flash_fwd_mla_params &, cudaStream_t) [with Kernel_traits=Flash_fwd_kernel_traits_mla<576, 64, 64, 8, cutlass::bfloat16_t, 512>, SharedStorage=flash::SharedStorageMLA>]" at line 602
instantiation of "void run_mha_fwd_splitkv_mla(Flash_fwd_mla_params &, cudaStream_t) [with T=cutlass::bfloat16_t, Headdim=576]" at line 3 of D:\GitHub\deepseek-ai\FlashMLA\csrc\flash_fwd_mla_bf16_sm90.cu

D:\GitHub\deepseek-ai\FlashMLA\csrc\flash_fwd_mla_kernel.h(528): warning #221-D: floating-point value does not fit in required floating-point type
float global_lse = (sum_lse == 0.f || sum_lse != sum_lse) ? ((float)(1e+300)) : logf(sum_lse) + max_lse;
^
detected during:
instantiation of "void flash::flash_fwd_splitkv_mla_combine_kernel(Flash_fwd_mla_params) [with Element=cutlass::bfloat16_t, ElementAccum=float, index_t=int64_t, kHeadDimV=512, kMaxSplits=160]" at line 588
instantiation of "void run_flash_splitkv_fwd_mla(Flash_fwd_mla_params &, cudaStream_t) [with Kernel_traits=Flash_fwd_kernel_traits_mla<576, 64, 64, 8, cutlass::bfloat16_t, 512>, SharedStorage=flash::SharedStorageMLA>]" at line 602
instantiation of "void run_mha_fwd_splitkv_mla(Flash_fwd_mla_params &, cudaStream_t) [with T=cutlass::bfloat16_t, Headdim=576]" at line 3 of D:\GitHub\deepseek-ai\FlashMLA\csrc\flash_fwd_mla_bf16_sm90.cu

1 error detected in the compilation of "csrc/flash_fwd_mla_bf16_sm90.cu".
error: command 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.4\\bin\\nvcc.EXE' failed with exit code 4294967295

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.