added gpu tests, need to work on expanding python version range

Author: Ashutosh Tiwari

Makefile

@@ -31,3 +31,9 @@ format:
 	@poetry install --only lint
 	@poetry run black .
 	@poetry run pre-commit run --all-files
+
+.PHONY: run_tests
+run_tests:
+	@echo "running tests..."
+	@poetry install --only main --only test -vvv
+	@poetry run pytest -q tests

graph_ml/utils/config.py

@@ -0,0 +1,16 @@
+import os, sys
+import torch
+
+
+OS = os.name
+DEVICE_TYPE = 'cpu'
+if OS == 'posix':
+    if torch.cuda.is_available():
+        DEVICE_TYPE = 'cuda'
+    elif torch.backends.mps.is_available():
+        DEVICE_TYPE = 'mps'
+
+DEVICE = torch.device(DEVICE_TYPE)
+
+GPU_AVAILABLE = DEVICE_TYPE in ['cuda', 'mps']
+

graph_ml/utils/gpu_utils.py

@@ -0,0 +1,3 @@
+import torch
+
+OS = 

pyproject.toml

@@ -64,3 +64,8 @@ pytest = "^8.0.2"
 [tool.poetry.group.lint.dependencies]
 pre-commit = "^3.6.2"
 black = "^24.2.0"
+
+
+
+# poetry add git+ssh://[email protected]/thunderock/graph_ml.git
+# poetry add whl_url

tests/test_gpu_available.py

@@ -1,4 +1,25 @@
 from __future__ import annotations
+import os
 
 import pytest
-from graph_ml import add
+from graph_ml.utils import config
+
+OS = os.name
+
+def test_target_os():
+    assert OS == "posix"
+
+
+def test_device_type():
+    assert config.DEVICE_TYPE in ["cpu", "cuda", "mps"]
+
+def test_gpu_available():
+    if config.DEVICE_TYPE == "cuda":
+        assert config.GPU_AVAILABLE
+    elif config.DEVICE_TYPE == "mps":
+        assert config.GPU_AVAILABLE
+    elif config.DEVICE_TYPE == "cpu":
+        assert not config.GPU_AVAILABLE
+    else:
+        assert False
+