[mxfp8] bug fix and test updates related to new triton_calculate_scale param

[danielvegamyhre]

• Second call site with new triton_calculate_scale param needs to be updated, this slipped through in [mxfp8] support RCEIL in triton_to_mxfp8_dim0 kernel with inline PTX #3498, motivation for getting blackwells in CI soon!
• Update triton mx dim1 tests to test both floor and rceil. Even though we aren't using it right now, just for completeness

Tests

• pytest test/prototype/mx_formats/test_kernels.py verified whole test suite is passing locally this time

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/3522

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure, 1 Cancelled Job, 1 Unrelated Failure

As of commit dca46b4 with merge base 7035fb7 ():

NEW FAILURE - The following job has failed:

• Run Regression Tests (aarch64) / test-cpu-ops (macos-14) (gh)
  Process completed with exit code 1.

CANCELLED JOB - The following job was cancelled. Please retry:

• Run Regression Tests (aarch64) / test-cpu-ops (linux.arm64.2xlarge) (gh)
  ##[error]The operation was canceled.

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://downloa... / linux-job (gh) (trunk failure)
  test/test_low_bit_optim.py::TestFSDP2::test_fsdp2

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[danielvegamyhre]

cc @drisspg for review

[drisspg]

if we can we should try to have 1 default arg at the top user facing and then not populate for inner funcs should help with these types of bug

[danielvegamyhre]

the launcher functions do have defaults for scaling_mode (rceil) that are passed in to the actual kernels. (let me know if i misunderstood your comment)

[danielvegamyhre]

oh you mean in this torch reference impl? i will add a default.