ROCm / Tensile

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

Use conditional imports for new dependencies #1970

Closed bstefanuk closed 1 month ago

bstefanuk commented 1 month ago

This is a patch for resolving imports for rich and CSafeLoader/CSafeDumper---Python and OS-level dependencies, respectively.

Testing has demonstrated that, while Ubuntu 20.04 onwards comes equipped with appropriate libyaml development packages, RHEL and SLES don't necessarily. The changes in this PR add a ConditionalImports.py file so that that new dependencies can be added with stable fallbacks.

These changes are intended as a stopgap; a long-term solution may be to integrate the required dependencies into the CI staging images (in which case the changes in this PR may be reverted), if this is not plausible, then documenting steps for users to setup appropriate dependencies will be important.