Fix formatting issues

.github/check-spdx.yaml

@@ -0,0 +1,14 @@
+DEFAULT:
+  perform_check: yes  # Perform check for all files
+  allowed_licenses:
+    - Apache-2.0
+  license_for_new_files: Apache-2.0  # license to be used when inserting a new copyright notice
+  new_notice_c: |+  # notice for new C, CPP, H, HPP and LD files
+    // SPDX-FileCopyrightText: (c) {years} Tenstorrent AI ULC
+    //
+    // SPDX-License-Identifier: {license}
+
+  new_notice_python: |+  # notice for new python files
+    # SPDX-FileCopyrightText: (c) {years} Tenstorrent AI ULC
+    #
+    # SPDX-License-Identifier: {license}

.github/workflows/build-and-test.yml

@@ -60,7 +60,7 @@ jobs:
   #   - name: 'Tar install directory and metal lib directory'
   #     shell: bash
   #     working-directory: ${{ steps.strings.outputs.install-output-dir }}
-  #     run: | 
+  #     run: |
   #       tar cvf artifact.tar .
 
   #   - name: Upload install folder to archive
@@ -150,4 +150,3 @@ jobs:
         export LD_LIBRARY_PATH="/opt/ttmlir-toolchain/lib/:${{ steps.strings.outputs.install-output-dir }}/lib:${LD_LIBRARY_PATH}"
         source env/activate
         pytest -v test
-

.gitignore

@@ -1,2 +1 @@
 *.pyc
-

.pre-commit-config.yaml

@@ -0,0 +1,23 @@
+repos:
+  - repo: https://github.com/psf/black
+    rev: 22.3.0
+    hooks:
+      - id: black
+        language_version: python3
+  - repo: https://github.com/pre-commit/mirrors-clang-format
+    rev: v18.1.7
+    hooks:
+    - id: clang-format
+      types_or: [c++, c]
+      args: [-style=file, -i]
+  - repo: https://github.com/espressif/check-copyright/
+    rev: v1.0.3
+    hooks:
+      - id: check-copyright
+        args: ['--config', '.github/check-spdx.yaml']
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v3.4.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-added-large-files

env/activate

@@ -35,4 +35,3 @@ else
   export TT_METAL_LOGGER_LEVEL="ERROR"
 
 fi
-

requirements.txt

@@ -1,4 +1,4 @@
-torch-mlir 
+torch-mlir
 torch@https://download.pytorch.org/whl/cpu-cxx11-abi/torch-2.5.0%2Bcpu.cxx11.abi-cp311-cp311-linux_x86_64.whl
 torchvision
 --extra-index-url https://download.pytorch.org/whl/nightly/cpu

setup.py

@@ -1,32 +1,41 @@
+# SPDX-FileCopyrightText: (c) 2024 Tenstorrent AI ULC
+#
+# SPDX-License-Identifier: Apache-2.0
 import os
 import sys
 import subprocess
 from setuptools import setup, Extension
 from setuptools.command.build_ext import build_ext
 
+
 class CMakeBuild(build_ext):
     def build_extension(self, ext):
         extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
-        cfg = 'Debug' if self.debug else 'Release'
+        cfg = "Debug" if self.debug else "Release"
         cmake_args = [
-            f'-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}',
-            f'-DPYTHON_EXECUTABLE={sys.executable}',
-            f'-DCMAKE_BUILD_TYPE={cfg}',
+            f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}",
+            f"-DPYTHON_EXECUTABLE={sys.executable}",
+            f"-DCMAKE_BUILD_TYPE={cfg}",
         ]
-        build_args = ['--config', cfg]
+        build_args = ["--config", cfg]
 
         if not os.path.exists(self.build_temp):
             os.makedirs(self.build_temp)
-        subprocess.check_call(['cmake', ext.sourcedir] + cmake_args, cwd=self.build_temp)
-        subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp)
+        subprocess.check_call(
+            ["cmake", ext.sourcedir] + cmake_args, cwd=self.build_temp
+        )
+        subprocess.check_call(
+            ["cmake", "--build", "."] + build_args, cwd=self.build_temp
+        )
+
 
 setup(
-    name='tt_torch',
-    version='0.1',
-    author='Aleks Knezevic',
-    description='TT PyTorch FrontEnd',
-    long_description='',
-    ext_modules=[Extension('c_bindings', sources=[])],
-    cmdclass={'build_ext': CMakeBuild},
+    name="tt_torch",
+    version="0.1",
+    author="Aleks Knezevic",
+    description="TT PyTorch FrontEnd",
+    long_description="",
+    ext_modules=[Extension("c_bindings", sources=[])],
+    cmdclass={"build_ext": CMakeBuild},
     zip_safe=False,
-)
+)

test/conftest.py

@@ -5,6 +5,7 @@
 import pytest
 import torch
 
+
 @pytest.fixture(autouse=True)
 def run_around_tests():
     yield