Fix/no std runtime (#1100)

Author: nathanielsimard

crates/cubecl-cpu/Cargo.toml

@@ -23,11 +23,19 @@ mlir-dump = []
 
 
 [dependencies]
-cubecl-common = { path = "../cubecl-common", version = "=0.9.0-pre.5", default-features = false }
-cubecl-core = { path = "../cubecl-core", version = "=0.9.0-pre.5", default-features = false }
-cubecl-opt = { path = "../cubecl-opt", version = "=0.9.0-pre.5", default-features = false }
+cubecl-common = { path = "../cubecl-common", version = "=0.9.0-pre.5", default-features = false, features = [
+    "std",
+] }
+cubecl-core = { path = "../cubecl-core", version = "=0.9.0-pre.5", default-features = false, features = [
+    "std",
+] }
+cubecl-opt = { path = "../cubecl-opt", version = "=0.9.0-pre.5", default-features = false, features = [
+    "std",
+] }
 cubecl-runtime = { path = "../cubecl-runtime", version = "=0.9.0-pre.5", default-features = false, features = [
     "channel-mutex",
+    "std",
+    "storage-bytes",
 ] }
 cubecl-std = { path = "../cubecl-std", version = "=0.9.0-pre.5", default-features = false }
 

crates/cubecl-cpu/src/compute/server.rs

@@ -270,19 +270,21 @@ impl ComputeServer for CpuServer {
     }
 
     fn flush(&mut self, stream_id: StreamId) {
+        self.scheduler.execute_streams(vec![stream_id]);
         let stream = self.scheduler.stream(&stream_id);
         stream.flush();
     }
 
     fn sync(&mut self, stream_id: StreamId) -> DynFut<Result<(), ExecutionError>> {
-        self.utilities.logger.profile_summary();
+        self.scheduler.execute_streams(vec![stream_id]);
         let stream = self.scheduler.stream(&stream_id);
         stream.flush();
 
         Box::pin(async move { Ok(()) })
     }
 
     fn start_profile(&mut self, stream_id: StreamId) -> ProfilingToken {
+        self.scheduler.execute_streams(vec![stream_id]);
         let stream = self.scheduler.stream(&stream_id);
         stream.start_profile()
     }
@@ -292,7 +294,7 @@ impl ComputeServer for CpuServer {
         stream_id: StreamId,
         token: ProfilingToken,
     ) -> Result<ProfileDuration, ProfileError> {
-        self.utilities.logger.profile_summary();
+        self.scheduler.execute_streams(vec![stream_id]);
         let stream = self.scheduler.stream(&stream_id);
         stream.end_profile(token)
     }

crates/cubecl-runtime/src/lib.rs

@@ -15,7 +15,6 @@ pub mod id;
 pub mod kernel;
 
 /// Stream related utilities.
-#[cfg(feature = "std")]
 pub mod stream;
 
 /// Compute client module.

crates/cubecl-runtime/src/stream/base.rs

@@ -1,3 +1,4 @@
+use alloc::vec::Vec;
 use cubecl_common::stream_id::StreamId;
 
 /// Trait for creating streams, used by the stream pool to generate streams as needed.

crates/cubecl-runtime/src/stream/scheduler.rs

@@ -4,7 +4,7 @@ use crate::{
     server::Binding,
     stream::{StreamFactory, StreamPool},
 };
-use alloc::sync::Arc;
+use alloc::{format, sync::Arc, vec, vec::Vec};
 use cubecl_common::stream_id::StreamId;
 
 /// Defines a trait for a scheduler stream backend, specifying the types and behavior for task scheduling.