mirror of https://github.com/tracel-ai/burn.git
Cleanup
This commit is contained in:
parent
15f7745264
commit
01fe2d0ffa
|
@ -50,7 +50,7 @@ pub fn expand_sig(
|
|||
|
||||
let ident = &sig.ident;
|
||||
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()),
|
||||
};
|
||||
|
||||
|
|
|
@ -20,7 +20,7 @@ pub fn launch<R: Runtime>(device: &R::Device) {
|
|||
let input_handle = client.create(f32::as_bytes(input));
|
||||
let output_handle = client.empty(input.len() * core::mem::size_of::<f32>());
|
||||
|
||||
gelu_launch::<F32, R>(
|
||||
gelu::launch::<F32, R>(
|
||||
client.clone(),
|
||||
CubeCount::Static(1, 1, 1),
|
||||
CubeDim::default(),
|
||||
|
|
Loading…
Reference in New Issue