Add License headers (#87)

Author: msaroufim

.github/scripts/check_license_headers.py

@@ -0,0 +1,57 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
+#!/usr/bin/env python3
+
+import sys
+import argparse
+
+REQUIRED_LICENSE_TEXT = "Copyright (c) Meta Platforms, Inc. and affiliates."
+
+
+def check_license_header(file_path):
+    """Check if a Python file has the required license header."""
+    try:
+        with open(file_path, "r", encoding="utf-8") as f:
+            content = f.read()
+            return REQUIRED_LICENSE_TEXT in content
+    except Exception as e:
+        print(f"Error reading {file_path}: {e}")
+        return False
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Check license headers in Python files")
+    parser.add_argument("files", nargs="*", help="Files to check")
+    args = parser.parse_args()
+
+    if not args.files:
+        return 0
+
+    missing_headers = []
+
+    for file_path in args.files:
+        if file_path.endswith(".py"):
+            if not check_license_header(file_path):
+                missing_headers.append(file_path)
+
+    if missing_headers:
+        print("Missing license headers in the following files:")
+        for file_path in missing_headers:
+            print(f"  - {file_path}")
+        print("\nPlease add the following license header to the top of each file:")
+        print("# Copyright (c) Meta Platforms, Inc. and affiliates.")
+        print("# All rights reserved.")
+        print("#")
+        print("# This source code is licensed under the BSD 3-Clause license found in the")
+        print("# LICENSE file in the root directory of this source tree.")
+        return 1
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

.github/workflows/ruff.yml

@@ -26,4 +26,7 @@ jobs:
       run: uv run ruff check .
 
     - name: Run ruff format check
-      run: uv run ruff format --check . 
+      run: uv run ruff format --check .
+    
+    - name: Check license headers
+      run: python .github/scripts/check_license_headers.py $(find . -name "*.py" -type f -not -path "./.venv/*" -not -path "./__pycache__/*") 

.pre-commit-config.yaml

@@ -4,4 +4,12 @@ repos:
     hooks:
       - id: ruff
         args: [--fix]
-      - id: ruff-format
+      - id: ruff-format
+  - repo: local
+    hooks:
+      - id: check-license-headers
+        name: Check license headers
+        entry: python .github/scripts/check_license_headers.py
+        language: system
+        files: \.py$
+        pass_filenames: true

BackendBench/__init__.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
 """
 BackendBench: A PyTorch backend evaluation framework with monkey patching support.
 

BackendBench/backends/__init__.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
 """
 BackendBench backends submodule.
 

BackendBench/backends/aten.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
 from .base import Backend
 
 

BackendBench/backends/base.py

@@ -1,3 +1,10 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
+
 class Backend:
     def __init__(self, name):
         self.name = name

BackendBench/backends/directory.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
 import importlib.util
 import logging
 import os

BackendBench/backends/flag_gems.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
 import torch
 
 from BackendBench.opregistry import register_operator

BackendBench/backends/kernel_agent.py

@@ -1,3 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD 3-Clause license found in the
+# LICENSE file in the root directory of this source tree.
+
 import importlib.util
 import logging
 import os