Skip to content

Commit c2e48f3

Browse files
committed
divide out tensor engine tests (#649)
Summary: Pull Request resolved: #649 Rollback Plan: Differential Revision: D78989844
1 parent 42570c5 commit c2e48f3

File tree

9 files changed

+3
-6
lines changed

9 files changed

+3
-6
lines changed

.github/workflows/test-cpu.yml

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,4 @@ jobs:
3434
# Install the built wheel from artifact
3535
install_wheel_from_artifact
3636
37-
# Currently a no-op.
38-
# Tests requiring tensor engine / GPU need to be identified and flagged to skip.
39-
# We will just ensure monarch can be imported successfully.
40-
python -c "import monarch; print('Monarch imported successfully')"
37+
LC_ALL=C pytest python/tests/ -s -v -m "not oss_skip" --ignore=python/tests/tensor_engine -n 4

.github/workflows/test-cuda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,5 +55,5 @@ jobs:
5555
pyright python/tests/test_python_actors.py
5656
5757
# Run CUDA tests
58-
LC_ALL=C pytest python/tests/ -s -v -m "not oss_skip"
58+
LC_ALL=C pytest python/tests/tensor_engine -s -v -m "not oss_skip" -n 4
5959
python python/tests/test_mock_cuda.py

python/tests/test_coalescing.py renamed to python/tests/tensor_engine/test_coalescing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def local_device_mesh(
7878
num_hosts,
7979
gpu_per_host,
8080
activate,
81-
backend=str(backend_type),
81+
backend=backend_type.value,
8282
)
8383

8484
@property
File renamed without changes.

0 commit comments

Comments
 (0)