Skip to content

GitLab

Commit b31bfe72 authored by ljss's avatar ljss
Browse files

add missing copyright

parent 3e123bc9
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 0 deletions
csrc/flash_api.cpp
View file @ b31bfe72
// Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/csrc/flash_attn/flash_api.cpp // 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/python.h>
#include <torch/nn/functional.h> #include <torch/nn/functional.h>
......
csrc/softmax.h
View file @ b31bfe72
// Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/csrc/flash_attn/src/softmax.h // Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/csrc/flash_attn/src/softmax.h
/******************************************************************************
* Copyright (c) 2024, Tri Dao.
******************************************************************************/
#pragma once #pragma once
......
csrc/utils.h
View file @ b31bfe72
// Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/hopper/utils.h // Adapted from https://github.com/Dao-AILab/flash-attention/blob/main/hopper/utils.h
/******************************************************************************
* Copyright (c) 2024, Tri Dao.
******************************************************************************/
#pragma once #pragma once
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment