Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

pin GPU and use "--forked" for some tests #58

Merged
merged 6 commits into from
Feb 27, 2024
Merged

Conversation

andy-neuma
Copy link
Member

@andy-neuma andy-neuma commented Feb 26, 2024

SUMMARY:

  • update runner script to use --forked for tests in tests/distributed
  • enable all test points in test_basic_distributed_correctness.py
  • pin GPU while running kernel tests. using CUDA_VISIBLE_DEVICES=0 when running tests/kernels and tests/samplers

TEST PLAN:
runs on remote push

modulo formatting ... this gets us a bit further ...

https://neuralmagic.testmo.net/automation/runs/view/8887

.github/scripts/run-tests Outdated Show resolved Hide resolved
@andy-neuma andy-neuma requested a review from SageMoore February 27, 2024 18:44
Copy link

@SageMoore SageMoore left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think this is generally fine for now.

@andy-neuma andy-neuma merged commit 87861c8 into main Feb 27, 2024
2 checks passed
@andy-neuma andy-neuma deleted the multi-gpu-patches branch February 27, 2024 18:57
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants