Separate linting passes for PyTorch and JAX (#86)

Separate linting passes for different frameworks
Signed-off-by: Tim Moon <tmoon@nvidia.com>

qa/L0_jax_lint/CPPLINT.cfg

+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
+
+# Stop searching for additional config files.
+set noparent
+
+# Limit line length.
+linelength=100
+
+# Ignore the following errors.
+filter=-build/include_subdir
+filter=-build/namespaces
+filter=-readability/todo
+filter=-build/header_guard
+filter=-build/include

qa/L0_jax_lint/pylintrc

+[MASTER]
+extension-pkg-whitelist=transformer_engine_jax
+
+disable=too-many-locals,
+        invalid-name,
+        too-many-arguments,
+        abstract-method,
+        arguments-differ,
+        too-many-instance-attributes,
+        unsubscriptable-object,
+        import-outside-toplevel,
+        too-many-statements,
+        import-error,
+        too-many-lines,
+        use-maxsplit-arg,
+        protected-access,
+        pointless-string-statement,
+        cyclic-import,
+        duplicate-code,
+        no-member,
+        attribute-defined-outside-init,
+        global-statement,
+        too-many-branches,
+        global-variable-not-assigned,
+        redefined-argument-from-local

qa/L0_jax_lint/test.sh

+# Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+#
+# See LICENSE for license information.
+
+set -e
+
+: "${TE_PATH:=/opt/transformerengine}"
+
+pip install cpplint==1.6.0 pylint==2.13.5
+if [ -z "${PYTHON_ONLY}" ]
+then
+  cp $TE_PATH/qa/L0_jax_lint/CPPLINT.cfg $TE_PATH
+  cd $TE_PATH
+  echo "Checking common API headers"
+  cpplint --root transformer_engine/common/include --recursive transformer_engine/common/include
+  echo "Checking C++ files"
+  cpplint --recursive --exclude=transformer_engine/common/include transformer_engine/common
+  cpplint --recursive transformer_engine/jax
+fi
+if [ -z "${CPP_ONLY}" ]
+then
+  cp $TE_PATH/qa/L0_jax_lint/pylintrc $TE_PATH
+  cd $TE_PATH
+  echo "Checking Python files"
+  pylint --recursive=y transformer_engine/common transformer_engine/jax
+fi

qa/L0_lint/pylintrc

 [MASTER]
 extension-pkg-whitelist=torch,
-                        transformer_engine_extensions,
-                        transformer_engine_jax
+                        transformer_engine_extensions
 
 disable=too-many-locals,
         invalid-name,
@@ -29,4 +28,3 @@ disable=too-many-locals,
 [TYPECHECK]
 ignored-modules=torch
 ignored-classes=torch
-

qa/L0_lint/test.sh

@@ -9,16 +9,18 @@ set -e
 pip install cpplint==1.6.0 pylint==2.13.5
 if [ -z "${PYTHON_ONLY}" ]
 then
+  cp $TE_PATH/qa/L0_lint/CPPLINT.cfg $TE_PATH
+  cd $TE_PATH
   echo "Checking common API headers"
-  cd $TE_PATH && \
   cpplint --root transformer_engine/common/include --recursive transformer_engine/common/include
   echo "Checking C++ files"
-  cd $TE_PATH && \
   cpplint --recursive --exclude=transformer_engine/common/include transformer_engine
+  cpplint --recursive transformer_engine/pytorch
 fi
 if [ -z "${CPP_ONLY}" ]
 then
+  cp $TE_PATH/qa/L0_lint/pylintrc $TE_PATH
+  cd $TE_PATH
   echo "Checking Python files"
-  cd $TE_PATH && \
-  pylint --recursive=y transformer_engine
+  pylint --recursive=y transformer_engine/common transformer_engine/pytorch
 fi