diff --git a/csrc/flash_api.cpp b/csrc/flash_api.cpp
index d2567fe..9015735 100644
--- 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
index 4ab6ae9..17e293a 100644
--- 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
index 3b8dd52..50295f7 100644
--- 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