dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cleanup

This commit is contained in:
nathaniel 2024-07-11 17:27:47 -04:00
parent 15f7745264
commit 01fe2d0ffa
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
crates/burn-cube-macros/src/codegen_common/signature.rs
View File

@ -50,7 +50,7 @@ pub fn expand_sig(
let ident = &sig.ident; let ident = &sig.ident;
let ident = match mode { let ident = match mode {
ExpandMode::FuncImpl => syn::Ident::new(format!("__expand").as_str(), ident.span()), ExpandMode::FuncImpl => syn::Ident::new("__expand".to_string().as_str(), ident.span()),
_ => syn::Ident::new(format!("__expand_{ident}").as_str(), ident.span()), _ => syn::Ident::new(format!("__expand_{ident}").as_str(), ident.span()),
}; };

2
examples/gelu/src/lib.rs
View File

@ -20,7 +20,7 @@ pub fn launch<R: Runtime>(device: &R::Device) {
let input_handle = client.create(f32::as_bytes(input)); let input_handle = client.create(f32::as_bytes(input));
let output_handle = client.empty(input.len() * core::mem::size_of::<f32>()); let output_handle = client.empty(input.len() * core::mem::size_of::<f32>());
gelu_launch::<F32, R>( gelu::launch::<F32, R>(
client.clone(), client.clone(),
CubeCount::Static(1, 1, 1), CubeCount::Static(1, 1, 1),
CubeDim::default(), CubeDim::default(),