huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 02:16:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump the ug dependency. (#2720)

Browse Source 
* Bump the ug dependency.

* Fix some test.

* Fix the ug test.
This commit is contained in:
Laurent Mazare
2025-01-16 09:39:16 +01:00
committed by GitHub
parent efd0e6822f
commit 6fd2f63a15
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/tests/custom_op_tests.rs
View File

@ -158,7 +158,7 @@ fn ug_op() -> Result<()> {
let st = op::store(ptr.id(), layout, src)?;
let kernel = op::Kernel::new("exp".to_string(), vec![ptr], vec![st]);
let opts: ug::lower_op::Opts = Default::default();
kernel.lower(&opts.with_global(0, 12))?
kernel.lower(&opts)?
};
let device = if candle_core::utils::cuda_is_available() {
Device::new_cuda(0)?