Skip to content

Fix output_type=BF16 test_backward_adagrad_large_dims test#151

Open
avbokovoy wants to merge 1 commit intoabokovoi/upstreamfrom
abokovoi/fix-bf16-bwd-adagrad-large-dims-test
Open

Fix output_type=BF16 test_backward_adagrad_large_dims test#151
avbokovoy wants to merge 1 commit intoabokovoi/upstreamfrom
abokovoi/fix-bf16-bwd-adagrad-large-dims-test

Conversation

@avbokovoy
Copy link

Currently, vec4acc.cuh stores don't account for bf16 outputs, treating them as FP16 during conversion from FP32. That makes test_backward_adagrad_large_dims test to fail in some specific cases when D>1024 (or D>256 in the context of unit-test parameters) and forward_v2 kernel is called. The way to reproduce the failure is to add @reproduce_failure('6.151.9', b'AEEAAEEAQQFCAQFBAUEDQQFBAQAAAAAAQQBBAg==') decorator to test_backward_adagrad_large_dims test and run on current main branch.

@avbokovoy avbokovoy self-assigned this Mar 25, 2026
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