search

ROCm / Tensile

Stretching GPU performance for GEMMs and tensor contractions.
MIT License
208 stars 142 forks source link

Filter kernels with build errors #1966

Closed bstefanuk closed 1 month ago

bstefanuk commented 1 month ago

Simple refactor:

  1. Add a function in TensileCreateLibrary.py to filter kernels that emit and error when building (relates to kernel writing logic).
  2. Add isort to automatically sort import for select files—reduces merge conflicts.
  3. Update printExit to display red.

Testing

Unit tests are added for the new function and existing related functions.

Notable changes

  1. The TensileCreateLibrary import in test_TensileCreateLibrary is renamed to tcl. So all functions should be accessed like tcl.functionName(...).
  2. An import sorting check has been added to CI, so make sure to run tox run -m static prior to committing to ensure formatting, linting, and import sorting are all complete.