adding precommit hook

Author: fschlimb

.pre-commit-config.yaml

@@ -0,0 +1,18 @@
+# See https://pre-commit.com for more information
+# See https://pre-commit.com/hooks.html for more hooks
+repos:
+-   repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.4.0
+    hooks:
+    -   id: end-of-file-fixer
+    -   id: trailing-whitespace
+-   repo: https://github.com/pocc/pre-commit-hooks
+    rev: v1.3.5
+    hooks:
+    -   id: clang-format
+        args: ["-i"]
+-   repo: https://github.com/psf/black
+    rev: 23.1.0
+    hooks:
+    - id: black
+      language_version: python3.9

CMakeLists.txt

@@ -44,7 +44,7 @@ set(CMAKE_CXX_STANDARD_REQUIRED ON)
 
 # ===============
 # Deps
-# =============== 
+# ===============
 # Find dependent packages like IMEX, Python3 and NumPy
 find_package(Python3 COMPONENTS Interpreter Development.Module NumPy REQUIRED)
 find_package(pybind11 CONFIG)

README.md

@@ -57,4 +57,11 @@ python DDPT_IDTR_SO=`pwd`/ddptensor/libidtr.so python simple.py
 and multi-process run is executed like
 ```bash
 python DDPT_IDTR_SO=`pwd`/ddptensor/libidtr.so mpirun -n 5 python simple.py
-```
+```
+
+## Contributing
+Please setup precommit hooks like this
+```
+pre-commit install -f -c ./.pre-commit-config.yaml
+pre-commit autoupdate
+```

ddptensor/__init__.py

@@ -20,7 +20,7 @@
     FLOAT32 as float32,
     INT64 as int64,
     INT32 as int32,
-    INT16  as int16,
+    INT16 as int16,
     INT8 as int8,
     UINT64 as uint64,
     UINT32 as uint32,
@@ -29,23 +29,26 @@
     BOOL as bool,
     init as _init,
     fini,
-    sync
+    sync,
 )
 
 from .ddptensor import dtensor
 from os import getenv
 from . import array_api as api
 from . import spmd
 
-_ddpt_cw = _bool(int(getenv('DDPT_CW', True)))
+_ddpt_cw = _bool(int(getenv("DDPT_CW", True)))
+
 
 def init(cw=None):
     cw = _ddpt_cw if cw is None else cw
     _init(cw)
 
+
 def to_numpy(a):
     return _cdt.to_numpy(a._t)
 
+
 for op in api.api_categories["EWBinOp"]:
     if not op.startswith("__"):
         OP = op.upper()
@@ -56,9 +59,7 @@ def to_numpy(a):
 for op in api.api_categories["EWUnyOp"]:
     if not op.startswith("__"):
         OP = op.upper()
-        exec(
-            f"{op} = lambda this: dtensor(_cdt.EWUnyOp.op(_cdt.{OP}, this._t))"
-        )
+        exec(f"{op} = lambda this: dtensor(_cdt.EWUnyOp.op(_cdt.{OP}, this._t))")
 
 for func in api.api_categories["Creator"]:
     FUNC = func.upper()
@@ -98,7 +99,10 @@ def to_numpy(a):
 
 for func in api.api_categories["LinAlgOp"]:
     FUNC = func.upper()
-    if func in ["tensordot", "vecdot",]:
+    if func in [
+        "tensordot",
+        "vecdot",
+    ]:
         exec(
             f"{func} = lambda this, other, axis: dtensor(_cdt.LinAlgOp.{func}(this._t, other._t, axis))"
         )
@@ -107,9 +111,7 @@ def to_numpy(a):
             f"{func} = lambda this, other: dtensor(_cdt.LinAlgOp.vecdot(this._t, other._t, 0))"
         )
     elif func == "matrix_transpose":
-        exec(
-            f"{func} = lambda this: dtensor(_cdt.LinAlgOp.{func}(this._t))"
-        )
+        exec(f"{func} = lambda this: dtensor(_cdt.LinAlgOp.{func}(this._t))")
 
 for func in api.api_categories["SortOp"]:
     exec(