Fix scale dtype and refactor q_dot_dq

[wenscarl]

1. Address type mismatches or inconsistencies in scale factors for FP8 operations.
2. Refactor the q-dot-dq (direct quantization pattern) into smaller, more granular components to enable greater flexibility.
   @kaixih @levskaya
   Support fp8 direct quantization praxis#69 depends on this PR.

[levskaya]

I'm not able to do a deep review of the fp8 quant details, but everything looks fine on the surface - you just need to correct one trailing space issue that's blocking our formatting presubmits.

I can't tell if this is ready to go in, but if it is let me know and I can merge it in!

[wenscarl]

I'm not able to do a deep review of the fp8 quant details, but everything looks fine on the surface - you just need to correct one trailing space issue that's blocking our formatting presubmits.

I can't tell if this is ready to go in, but if it is let me know and I can merge it in!

Thanks for reviewing it. The formatting is fixed. It's ready to go in from our e2e test.

[levskaya]

@wenscarl - I'm seeing actual test failures here? is it just a tolerance issue or something more serious?

[wenscarl]

@levskaya It was a typo and fixed. All tests passed.