Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

hotfix: bugfix to #756 #757

Merged
merged 1 commit into from
Jan 27, 2025
Merged

hotfix: bugfix to #756 #757

merged 1 commit into from
Jan 27, 2025

Conversation

yzh119
Copy link
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
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