Make cuda is configurable. #66

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.

Sign up for GitHub

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

Jump to bottom

Draft

AlongWY wants to merge 5 commits into nbigaouette:master from AlongWY:al/cuda

AlongWY commented Feb 28, 2021 •

edited

Loading

Done

add includes in wrapper.h (with #ifdefine)
use system env to control the define in onnxruntime-sys build.rs
add cuda and cpu provider in session
update examples

Test

Macos
Linux with cuda 10.2 and cudnn 8.0

Todo

regenerate bindings

AlongWY added 5 commits

February 28, 2021 11:25


          add support to cuda

041cc49


          fix format

a0bdfbd


          fix cuda DEFINE

641024d


          update sample.rs

9f63b2f


          remove unused mut

d094718

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet