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

Cleanup the pyo3 setup. (#705)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-01 15:26:18 +02:00
committed by GitHub
parent af552a5274
commit 237323c2bc
3 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
Makefile
View File

@ -1,3 +1,15 @@
pyo3-test:
cargo build --profile=release-with-debug --package candle-pyo3
cp -f ./target/release-with-debug/libcandle.so candle.so
PYTHONPATH=.:$PYTHONPATH python3 candle-pyo3/test.py
pyo3-test-macos:
cargo build --profile=release-with-debug --package candle-pyo3
cp -f ./target/release-with-debug/libcandle.dylib candle.so
PYTHONPATH=.:$PYTHONPATH python3 candle-pyo3/test.py
.PHONY: pyo3-test pyo3-test-macos clean-ptx clean test
clean-ptx: clean-ptx:
find target -name "*.ptx" -type f -delete find target -name "*.ptx" -type f -delete
echo "" > candle-kernels/src/lib.rs echo "" > candle-kernels/src/lib.rs
@ -11,8 +23,4 @@ clean:
test: test:
cargo test cargo test
pyo3-test:
cargo build --profile=release-with-debug --package candle-pyo3
python3 candle-pyo3/test.py
all: test all: test

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

@ -16,7 +16,6 @@ pub enum Device {
Cuda(crate::CudaDevice), Cuda(crate::CudaDevice),
} }
// TODO: Should we back the cpu implementation using the NdArray crate or similar?
pub trait NdArray { pub trait NdArray {
fn shape(&self) -> Result<Shape>; fn shape(&self) -> Result<Shape>;

1
candle-pyo3/src/lib.rs
View File

@ -470,6 +470,7 @@ fn candle(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
m.add_class::<PyDType>()?; m.add_class::<PyDType>()?;
m.add("u8", PyDType(DType::U8))?; m.add("u8", PyDType(DType::U8))?;
m.add("u32", PyDType(DType::U32))?; m.add("u32", PyDType(DType::U32))?;
m.add("i16", PyDType(DType::I64))?;
m.add("bf16", PyDType(DType::BF16))?; m.add("bf16", PyDType(DType::BF16))?;
m.add("f16", PyDType(DType::F16))?; m.add("f16", PyDType(DType::F16))?;
m.add("f32", PyDType(DType::F32))?; m.add("f32", PyDType(DType::F32))?;