ROCm / Tensile

Stretching GPU performance for GEMMs and tensor contractions.
MIT License
220 stars 149 forks source link

Patch tuning imports for PyYAML #1979

Closed bstefanuk closed 3 months ago

bstefanuk commented 3 months ago

A recent patch added a dependency on Tensile in some tuning scripts, causing the following error:

Traceback (most recent call last):
File "Tensile/tuning/automation/rocblas-benchInputCreator.py", line 47, in <module>
from Tensile.Utilities.ConditionalImports import yamlLoader, yamlDumper
ModuleNotFoundError: No module named 'Tensile'

This PR resolves this error by replacing the PyYAML import directly.

Since there aren't any tests in CI for verifying manual workflows, going forward the Tensile Frameworks team will focus only on updates to the core Tensile module.