add triton to uv

Signed-off-by: Hao Wu <[email protected]>

Author: skyw

pyproject.toml

@@ -84,6 +84,7 @@ dev = [
     "pre-commit>=3.6.0",
     "ruff>=0.9.9",
     "mypy>=1.8.0",
+    "triton>=3.4.0",
 ]
 
 [tool.uv]

tests/ci/L0_Tests_CPU.sh

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 export TORCH_COMPILE_DISABLE=1
-error=0 
+
+error=0
 torchrun --nproc_per_node=8 --no-python coverage run -p tests/test_distributed_muon_utils_cpu.py || error=1
 torchrun --nproc_per_node=4 --no-python coverage run -p tests/test_distributed_muon_utils_cpu.py || error=1
 coverage run -p --source=emerging_optimizers tests/test_scalar_optimizers.py --device=cpu || error=1

tests/ci/L0_Tests_GPU.sh

@@ -15,8 +15,7 @@ export TORCH_COMPILE_DISABLE=1
 export CUDA_VISIBLE_DEVICES=0
 export TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=0
 
-
-error=0 
+error=0
 coverage run -p --source=emerging_optimizers tests/test_muon_utils.py || error=1
 coverage run -p --source=emerging_optimizers tests/test_orthogonalized_optimizer.py || error=1
 coverage run -p --source=emerging_optimizers tests/test_soap_functions.py || error=1

tests/ci/L1_Tests_GPU.sh

@@ -14,7 +14,7 @@
 export CUDA_VISIBLE_DEVICES=0
 export TORCH_ALLOW_TF32_CUBLAS_OVERRIDE=0
 
-error=0 
+error=0
 python tests/test_muon_utils.py || error=1
 python tests/test_orthogonalized_optimizer.py || error=1
 python tests/test_soap_functions.py || error=1