Skip to content

[ROCm] [TunableOp] Enable logging of BLAS parameters#147034

Closed
naromero77amd wants to merge 23 commits intopytorch:mainfrom
ROCm:tunableop_offline_blas_log
Closed

[ROCm] [TunableOp] Enable logging of BLAS parameters#147034
naromero77amd wants to merge 23 commits intopytorch:mainfrom
ROCm:tunableop_offline_blas_log

Conversation

@naromero77amd
Copy link
Collaborator

@naromero77amd naromero77amd commented Feb 12, 2025

This PR supports a logging feature that is being requested.

PYTORCH_TUNABLEOP_BLAS_LOG=1

Enables the logging of BLAS parameters with either offline or online (in-situ) tuning.

The BLAS parameters are written to the CSV file.

cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang

@pytorch-bot
Copy link

pytorch-bot bot commented Feb 12, 2025

🔗 Helpful Links

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

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

⏳ No Failures, 1 Pending

As of commit 7d01188 with merge base 703176e (image):
💚 Looks good so far! There are no failures yet. 💚

UNSTABLE - The following job is marked as unstable, possibly due to flakiness on trunk:

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

@pytorch-bot pytorch-bot bot added module: rocm AMD GPU support for Pytorch ciflow/rocm Trigger "default" config CI on ROCm labels Feb 12, 2025
@naromero77amd naromero77amd marked this pull request as draft February 12, 2025 23:31
@naromero77amd naromero77amd added the topic: not user facing topic category label Feb 12, 2025
@naromero77amd naromero77amd marked this pull request as ready for review February 19, 2025 22:35
return "Unknown ComputeType";
}

#ifdef USE_ROCM
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why is ComputeTypeFor inside the #ifdef USE_ROCM?

@mikaylagawarecki mikaylagawarecki added the triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module label Feb 25, 2025
@naromero77amd naromero77amd marked this pull request as draft March 4, 2025 20:30
@naromero77amd naromero77amd force-pushed the tunableop_offline_blas_log branch from 93db444 to 8655414 Compare March 4, 2025 22:40
@pytorch-bot pytorch-bot bot removed the ciflow/rocm Trigger "default" config CI on ROCm label Mar 5, 2025
@naromero77amd naromero77amd marked this pull request as ready for review March 6, 2025 20:47
@naromero77amd naromero77amd added the ciflow/rocm Trigger "default" config CI on ROCm label Mar 6, 2025
@pytorch-bot pytorch-bot bot removed the ciflow/rocm Trigger "default" config CI on ROCm label Mar 6, 2025
@naromero77amd naromero77amd added the ciflow/rocm Trigger "default" config CI on ROCm label Mar 6, 2025
@naromero77amd
Copy link
Collaborator Author

@pytorchbot merge

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Mar 7, 2025
@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

@naromero77amd naromero77amd deleted the tunableop_offline_blas_log branch March 7, 2025 14:10
pytorchmergebot pushed a commit that referenced this pull request Mar 11, 2025
…148979)

In a previous PR #147034, there was a bad merge at the last minute.
BLAS logging works for offline tuning, but does not currently work for online tuning.

This PR fixes BLAS logging for online tuning.

Pull Request resolved: #148979
Approved by: https://github.com/jeffdaily
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/rocm Trigger "default" config CI on ROCm ciflow/trunk Trigger trunk jobs on your pull request Merged module: rocm AMD GPU support for Pytorch open source topic: not user facing topic category triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants