DeepSeek/FlashMLA
1
0
Fork 0
mirror of https://github.com/deepseek-ai/FlashMLA synced 2025-05-03 03:32:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add missing copyright

Browse Source
This commit is contained in:
ljss 2025-03-01 18:24:24 +08:00
parent 3e123bc93c
commit b31bfe72a8
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
csrc/flash_api.cpp
View File

@ -1,4 +1,7 @@
// Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/csrc/flash_attn/flash_api.cpp
/******************************************************************************
* Copyright (c) 2024, Tri Dao.
******************************************************************************/
#include <torch/python.h>
#include <torch/nn/functional.h>

3
csrc/softmax.h
View File

@ -1,4 +1,7 @@
// Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/csrc/flash_attn/src/softmax.h
/******************************************************************************
* Copyright (c) 2024, Tri Dao.
******************************************************************************/
#pragma once

3
csrc/utils.h
View File

@ -1,4 +1,7 @@
// Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/hopper/utils.h
/******************************************************************************
* Copyright (c) 2024, Tri Dao.
******************************************************************************/
#pragma once