brainworm2024 commited on
Commit
feab073
·
1 Parent(s): 87840bf

Fix module structure: use library crate correctly

Browse files
Files changed (3) hide show
  1. Cargo.lock +0 -0
  2. src/lib.rs +2 -0
  3. src/main.rs +3 -9
Cargo.lock ADDED
The diff for this file is too large to render. See raw diff
 
src/lib.rs CHANGED
@@ -2,3 +2,5 @@ pub mod handlers;
2
  pub mod inference;
3
  pub mod shield;
4
  pub mod web3;
 
 
 
2
  pub mod inference;
3
  pub mod shield;
4
  pub mod web3;
5
+ pub mod orchestrator;
6
+ pub mod proof;
src/main.rs CHANGED
@@ -3,8 +3,8 @@
3
  //
4
  // To run on real MI300X GPUs:
5
  // 1. Set environment variable ENABLE_ROCM=1
6
- // 2. Ensure the ROCm runtime is installed (hipcc, rocblas, etc.)
7
- // 3. The model will automatically use Device::new_hip(0)
8
  // 4. The /status endpoint will show "ROCm/HIP (MI300X)"
9
  // ============================================================================
10
 
@@ -13,13 +13,7 @@ use serde::Serialize;
13
  use tower_http::trace::TraceLayer;
14
  use tracing_subscriber::EnvFilter;
15
 
16
- mod handlers;
17
- mod inference;
18
- mod lib;
19
- mod shield;
20
- mod web3;
21
- mod orchestrator;
22
- mod proof;
23
 
24
  #[derive(Serialize)]
25
  struct StatusResponse {
 
3
  //
4
  // To run on real MI300X GPUs:
5
  // 1. Set environment variable ENABLE_ROCM=1
6
+ // 2. Ensure the ROCm runtime is installed
7
+ // 3. The model will use Device::new_hip(0)
8
  // 4. The /status endpoint will show "ROCm/HIP (MI300X)"
9
  // ============================================================================
10
 
 
13
  use tower_http::trace::TraceLayer;
14
  use tracing_subscriber::EnvFilter;
15
 
16
+ use rustvital_amd::handlers;
 
 
 
 
 
 
17
 
18
  #[derive(Serialize)]
19
  struct StatusResponse {