Skip to content

Add parameterized test for 2D tile sizes in QtProvider. - #340

Merged
copybara-service[bot] merged 1 commit into
mainfrom
test_951633206
Jul 21, 2026
Merged

Add parameterized test for 2D tile sizes in QtProvider.#340
copybara-service[bot] merged 1 commit into
mainfrom
test_951633206

Conversation

@copybara-service

Copy link
Copy Markdown

Add parameterized test for 2D tile sizes in QtProvider.

This change adds a new parameterized test, test_nnx_linear_qt_2d_bwd, to verify that QtProvider is compatible with 2D tile sizes. The test covers various combinations of 2D and scalar tile sizes for weights, backward weight gradients, dlhs, and drhs. To support this, QtTest is updated to inherit from parameterized.TestCase.

@copybara-service
copybara-service Bot force-pushed the test_951633206 branch 3 times, most recently from 5172f52 to 0e4d858 Compare July 21, 2026 22:59
This change adds a new parameterized test, test_nnx_linear_qt_2d_bwd, to verify that QtProvider is compatible with 2D tile sizes. The test covers various combinations of 2D and scalar tile sizes for weights, backward weight gradients, dlhs, and drhs. To support this, QtTest is updated to inherit from parameterized.TestCase.

PiperOrigin-RevId: 951749225
@copybara-service
copybara-service Bot merged commit 41cdd7c into main Jul 21, 2026
@copybara-service
copybara-service Bot deleted the test_951633206 branch July 21, 2026 23:02
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