Updated tox and pyproject

Signed-off-by: romitjain <[email protected]>

Author: romitjain

pyproject.toml

@@ -52,6 +52,14 @@ fms-accel = ["fms-acceleration>=0.6.2"]
 gptq-dev = ["auto_gptq>0.4.2", "optimum>=1.15.0"]
 mamba = ["mamba_ssm[causal-conv1d]>=2.0.0,<3.0.0"]
 scanner-dev = ["HFResourceScanner>=0.1.0"]
+fms-accel-all = [
+    "fms-acceleration>=0.6.2",
+    "fms-acceleration-peft",
+    "fms-acceleration-foak",
+    "fms-acceleration-aadp",
+    "fms-acceleration-moe",
+    "fms-acceleration-odm"
+]
 
 [tool.setuptools.packages.find]
 exclude = ["tests", "tests.*"]

tox.ini

@@ -53,3 +53,14 @@ commands =
     coverage report -m
     coverage xml
     genbadge coverage -s -i coverage.xml
+
+[testenv:accel]
+description = run GPU enabled tests
+deps = 
+    pytest>=7
+    .[aim,mlflow,clearml,scanner-dev,fms-accel-all]
+setenv =
+    CUDA_VISIBLE_DEVICES=0
+commands =
+    pytest {posargs:tests/test_sft_trainer.py}
+    pytest {posargs:tests/acceleration/test_acceleration_framework.py}