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

Remove tracing.

Browse Source
This commit is contained in:
Nicolas Patry
2023-11-09 19:41:08 +01:00
parent 8cf39d27ce
commit af5e77f409
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-core/Cargo.toml
View File

@ -12,7 +12,6 @@ readme = "README.md"
[dependencies] [dependencies]
accelerate-src = { workspace = true, optional = true } accelerate-src = { workspace = true, optional = true }
byteorder = { workspace = true } byteorder = { workspace = true }
tracing = { workspace = true }
candle-kernels = { path = "../candle-kernels", version = "0.3.0", optional = true } candle-kernels = { path = "../candle-kernels", version = "0.3.0", optional = true }
metal = { workspace = true, optional = true} metal = { workspace = true, optional = true}
cudarc = { workspace = true, optional = true } cudarc = { workspace = true, optional = true }

12
candle-core/src/device.rs
View File

@ -1,6 +1,6 @@
use crate::backend::BackendDevice; use crate::backend::BackendDevice;
use crate::cpu_backend::CpuDevice; use crate::cpu_backend::CpuDevice;
use crate::{bail, CpuStorage, DType, Result, Shape, Storage, WithDType}; use crate::{CpuStorage, DType, Result, Shape, Storage, WithDType};
/// A `DeviceLocation` represents a physical device whereas multiple `Device` /// A `DeviceLocation` represents a physical device whereas multiple `Device`
/// can live on the same location (typically for cuda devices). /// can live on the same location (typically for cuda devices).
@ -105,14 +105,14 @@ impl<S: WithDType, const N1: usize, const N2: usize, const N3: usize, const N4:
impl<S: NdArray> NdArray for Vec<S> { impl<S: NdArray> NdArray for Vec<S> {
fn shape(&self) -> Result<Shape> { fn shape(&self) -> Result<Shape> {
if self.is_empty() { if self.is_empty() {
bail!("empty array") crate::bail!("empty array")
} }
let shape0 = self[0].shape()?; let shape0 = self[0].shape()?;
let n = self.len(); let n = self.len();
for v in self.iter() { for v in self.iter() {
let shape = v.shape()?; let shape = v.shape()?;
if shape != shape0 { if shape != shape0 {
bail!("two elements have different shapes {shape:?} {shape0:?}") crate::bail!("two elements have different shapes {shape:?} {shape0:?}")
} }
} }
Ok(Shape::from([[n].as_slice(), shape0.dims()].concat())) Ok(Shape::from([[n].as_slice(), shape0.dims()].concat()))
@ -166,6 +166,10 @@ impl Device {
matches!(self, Self::Cuda(_)) matches!(self, Self::Cuda(_))
} }
pub fn is_metal(&self) -> bool {
matches!(self, Self::Metal(_))
}
pub fn cuda_if_available(ordinal: usize) -> Result<Self> { pub fn cuda_if_available(ordinal: usize) -> Result<Self> {
if crate::utils::cuda_is_available() { if crate::utils::cuda_is_available() {
Self::new_cuda(ordinal) Self::new_cuda(ordinal)
@ -199,7 +203,7 @@ impl Device {
Device::Metal(_device) => { Device::Metal(_device) => {
// let storage = device.rand_uniform(shape, dtype, lo, up)?; // let storage = device.rand_uniform(shape, dtype, lo, up)?;
// Ok(Storage::Metal(storage)) // Ok(Storage::Metal(storage))
bail!("Metal rand_uniform not implemented") crate::bail!("Metal rand_uniform not implemented")
} }
} }
} }