diff --git a/deepspeed/runtime/zero/compile/passes/offload_activation.py b/deepspeed/runtime/zero/compile/passes/offload_activation.py
index 47579c990a62..2f0139a0f73f 100644
--- a/deepspeed/runtime/zero/compile/passes/offload_activation.py
+++ b/deepspeed/runtime/zero/compile/passes/offload_activation.py
@@ -1,3 +1,8 @@
+# Copyright (c) Microsoft Corporation.
+# SPDX-License-Identifier: Apache-2.0
+
+# DeepSpeed Team
+
 from typing import List, Dict, Set, Tuple
 import random
 from collections import defaultdict