Skip to content

hotfix: bugfix to #756#757

Merged
yzh119 merged 1 commit into
mainfrom
hotfix-756
Jan 27, 2025
Merged

hotfix: bugfix to #756#757
yzh119 merged 1 commit into
mainfrom
hotfix-756

Conversation

@yzh119
Copy link
Copy Markdown
Collaborator

@yzh119 yzh119 commented Jan 27, 2025

This pull request includes changes to the flashinfer/jit/__init__.py file to improve the import structure and handle the prebuilt_ops_uri import conditionally.

Improvements to import structure:

  • Removed the unconditional import of prebuilt_ops_uri from aot_config and added it conditionally within the try-except block to handle cases where _kernels or _kernels_sm90 are not available. [1] [2]

cc @ByronHsu

@ByronHsu
Copy link
Copy Markdown
Collaborator

Thanks!

@yzh119 yzh119 merged commit 68d1177 into main Jan 27, 2025
@zhyncs zhyncs deleted the hotfix-756 branch January 28, 2025 03:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants