Compare commits

...

6 Commits

Author SHA1 Message Date
8f920b4799 Update on "bf16 support for per_channel bwd"
Follow up to #165098 - adding bf16 support for the backward pass. To avoid BC breaking changes/losing precision, we upcast the parameters to fp32 after the op gets called, and downcast the gradients to bf16 before returning.

For testing, we upcast to fp32 before calling the reference function.

[ghstack-poisoned]
2025-10-14 11:51:45 -07:00
7721d5a806 Update on "bf16 support for per_channel bwd"
Follow up to #165098 - adding bf16 support for the backward pass. To avoid BC breaking changes/losing precision, we upcast the parameters to fp32 after the op gets called, and downcast the gradients to bf16 before returning.

For testing, we upcast to fp32 before calling the reference function.

[ghstack-poisoned]
2025-10-14 11:51:28 -07:00
e51552344e Update on "bf16 support for per_channel bwd"
Follow up to #165098 - adding bf16 support for the backward pass. To avoid BC breaking changes/losing precision, we upcast the parameters to fp32 after the op gets called, and downcast the gradients to bf16 before returning.

For testing, we upcast to fp32 before calling the reference function.

[ghstack-poisoned]
2025-10-14 11:51:02 -07:00
a12e2d1296 Update on "bf16 support for per_channel bwd"
[ghstack-poisoned]
2025-10-14 11:49:14 -07:00
04b5566455 Update on "bf16 support for per_channel bwd"
[ghstack-poisoned]
2025-10-14 11:49:11 -07:00
b014c1a894 bf16 support for per_channel bwd
[ghstack-poisoned]
2025-10-14 11:48:04 -07:00

Diff Content Not Available