Skip to content

[tests] tighten keep_in_32 modules tests - #14399

Open
sayakpaul wants to merge 3 commits into
mainfrom
tighten-keep32-modules
Open

[tests] tighten keep_in_32 modules tests#14399
sayakpaul wants to merge 3 commits into
mainfrom
tighten-keep32-modules

Conversation

@sayakpaul

@sayakpaul sayakpaul commented Aug 6, 2026

Copy link
Copy Markdown
Member

What does this PR do?

As the title suggests. _keep_in_fp32_modules is an important model class-level attribute that deserves better functional testing. In particular, this PR adds / modifies the following:

  • Removes the require_accelerator marker from test_keep_in_fp32_modules because there's no computation being run there. CPU is enough for that test.
  • Layerwise casting + _keep_in_fp32_modules

@github-actions github-actions Bot added tests size/M PR with diff < 200 LOC labels Aug 6, 2026
@sayakpaul
sayakpaul requested a review from DN6 August 6, 2026 15:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

size/M PR with diff < 200 LOC tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant