Skip to content
This repository was archived by the owner on Nov 3, 2023. It is now read-only.

[cuda ext] Protect Cuda Extension Loading #4779

Merged
merged 1 commit into from
Sep 9, 2022

Conversation

klshuster
Copy link
Contributor

Patch description
Wrap loading of our cuda extension for GPU ngram blocking in a try/catch.

Testing steps
YOLO

Copy link
Contributor

@mojtaba-komeili mojtaba-komeili left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for fixing it.

@klshuster klshuster merged commit e217dc4 into main Sep 9, 2022
@klshuster klshuster deleted the protect_ngram_cuda_loading branch September 9, 2022 15:47
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants