Skip to content

Conversation

@zahiqbal
Copy link
Contributor

…ction stage

(cherry picked from commit 42b19d0)

Motivation

This PR fixes pytest "test collection" and execution mechanism. Test are collected in stage one after filtering out unwanted tests., and passed to 2nd stage for execution.

However Multigpu tests are not following test collection mechanism. It directly run tests files (.py) with mutligpu parameterized tests variants.
In this case, a python dictionary is introduced that list all unwanted tests in corresponding test file. this list is used to filter out unwanted test list during test-file execution.

Technical Details

Test Plan

JAX Unit test

Test Result

PASSED

Submission Checklist

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant