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

Updated everything and output a trace.

Browse Source
This commit is contained in:
Nicolas Patry
2023-11-07 21:06:20 +01:00
parent 76d3116f5d
commit 0c24a885a6
8 changed files with 249 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
candle-examples/examples/quantized/main.rs
View File

@ -238,13 +238,16 @@ fn main() -> anyhow::Result<()> {
} else {
Some(args.temperature)
};
let _guard = if args.tracing {
let (chrome_layer, guard) = ChromeLayerBuilder::new().build();
tracing_subscriber::registry().with(chrome_layer).init();
Some(guard)
} else {
None
};
tracing_subscriber::fmt::init();
// let _guard = if args.tracing {
// // let (chrome_layer, guard) = ChromeLayerBuilder::new().build();
// // tracing_subscriber::registry().with(chrome_layer).init();
// tracing_subscriber::fmt::init();
// None
// // Some(guard)
// } else {
// None
// };
println!(
"avx: {}, neon: {}, simd128: {}, f16c: {}",