add missing copyright

beginlner · beginlner · commit b31bfe72a83e · 2025-03-01T18:24:24.000+08:00
diff --git a/csrc/flash_api.cpp b/csrc/flash_api.cpp
@@ -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>
diff --git a/csrc/softmax.h b/csrc/softmax.h
@@ -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
 
diff --git a/csrc/utils.h b/csrc/utils.h
@@ -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
 
