Skip to content

Commit

Permalink
fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
JinWeiTan committed Nov 3, 2023
1 parent f57f42f commit b270f5e
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 6 deletions.
2 changes: 1 addition & 1 deletion crates/core-gpu/src/ffi.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ use std::slice::{from_raw_parts, from_raw_parts_mut};

use crate::{
decode_array, decode_json, length, Backend, Dataset, Logger, PredictOptions, TrainOptions,
RESOURCES, WGPUBackend,
WGPUBackend, RESOURCES,
};

type AllocBufferFn = extern "C" fn(usize) -> *mut u8;
Expand Down
2 changes: 1 addition & 1 deletion crates/core-gpu/src/gpu/layers/dense.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
use ndarray::{Array1, Array2, ArrayD, Axis, Dimension, Ix1, Ix2, IxDyn};
use std::ops::Add;

use crate::{GPUInit, DenseLayer, Init, Tensors};
use crate::{DenseLayer, GPUInit, Init, Tensors};

pub struct DenseGPULayer {
// cache
Expand Down
4 changes: 2 additions & 2 deletions crates/core-gpu/src/gpu/mod.rs
Original file line number Diff line number Diff line change
@@ -1,17 +1,17 @@
mod activation;
mod backend;
mod cost;
mod gpu;
mod init;
mod layers;
mod optimizers;
mod schedulers;
mod gpu;

pub use activation::*;
pub use backend::*;
pub use cost::*;
pub use gpu::*;
pub use init::*;
pub use layers::*;
pub use optimizers::*;
pub use schedulers::*;
pub use gpu::*;
2 changes: 1 addition & 1 deletion crates/core-gpu/src/lib.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
mod gpu;
mod ffi;
mod gpu;
mod tensor;
mod types;
mod util;
Expand Down
1 change: 0 additions & 1 deletion crates/core/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@ mod wasm;

pub use cpu::*;


#[cfg(not(target_arch = "wasm32"))]
pub use ffi::*;
pub use tensor::*;
Expand Down

0 comments on commit b270f5e

Please sign in to comment.