huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add CANDLE_NVCC_CCBIN support for candle-kernels, and eliminate warning. (#836)

Browse Source
This commit is contained in:
Charles Lew
2023-09-13 18:39:22 +08:00
committed by GitHub
parent 3e94324012
commit 1c09164021
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/src/cuda_backend.rs
View File

@ -12,6 +12,7 @@ use half::{bf16, f16};
use std::sync::{Arc, Mutex};
const USE_IM2COL_CONV1D: bool = true;
#[cfg(not(feature = "cudnn"))]
const USE_IM2COL_CONV2D: bool = true;
/// cudarc related errors