Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
41 commits
Select commit Hold shift + click to select a range
1f4a0ad
Tested virtualstorage
RaulMoldes Sep 20, 2025
013421b
Need to clean it up and test, but a good first approximation
RaulMoldes Sep 21, 2025
acca686
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 21, 2025
64ebfdc
Need to clean it up and test, but a good first approximation
RaulMoldes Sep 21, 2025
2495386
Fixed clippy warnings
RaulMoldes Sep 22, 2025
c1ee1c6
Fixed clippy warnings
RaulMoldes Sep 22, 2025
e81d440
Implemented a first approximation of defragmentation
RaulMoldes Sep 22, 2025
152cc3a
Implemented a first approximation of defragmentation
RaulMoldes Sep 22, 2025
ee2da4c
Implemented a first approximation of defragmentation
RaulMoldes Sep 22, 2025
57ea278
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 22, 2025
984ce86
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 22, 2025
510701b
merged remote
RaulMoldes Sep 22, 2025
4345474
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 23, 2025
79ce8a6
Added defragmentation tests
RaulMoldes Sep 23, 2025
69d835e
Need to think how to integrate with memory manager
RaulMoldes Sep 23, 2025
9016d85
Refactored GpVirtualStorage
RaulMoldes Sep 23, 2025
a3149be
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 23, 2025
36dad91
Refactored GpVirtualStorage
RaulMoldes Sep 23, 2025
6e5713c
Refactored GpVirtualStorage
RaulMoldes Sep 23, 2025
3fbbe50
configured prometheus with tsdb persistent storage
RaulMoldes Sep 23, 2025
586fda4
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 24, 2025
53651db
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 24, 2025
df46b90
Fixed clippy stuff and added memory pool
RaulMoldes Sep 24, 2025
99664eb
Added last_allocated_page to make contiguous page allocations
RaulMoldes Sep 24, 2025
fdf9e2f
Added cuda hints to allocate pages in a contiguous manner
RaulMoldes Sep 24, 2025
9dc68c3
I think i finished virtual memory pool
RaulMoldes Sep 25, 2025
d2b336c
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 25, 2025
a335614
Added testing in gpuvirtualstorage
RaulMoldes Sep 25, 2025
d81e498
Added testing macro
RaulMoldes Sep 25, 2025
3dca8d1
Merge remote-tracking branch 'upstream'
RaulMoldes Sep 25, 2025
65f9cfb
Added testing macro
RaulMoldes Sep 25, 2025
70495e5
added docs
RaulMoldes Sep 25, 2025
5677051
Added docs
RaulMoldes Sep 25, 2025
38602bc
Added docs
RaulMoldes Sep 25, 2025
cba13ad
tested first part
RaulMoldes Sep 25, 2025
316e9fd
Ready to test!
RaulMoldes Sep 26, 2025
4656a8d
Merge remote-tracking branch 'upstream/main'
RaulMoldes Sep 27, 2025
d3de680
Added test, cargo xtask validate is passing
RaulMoldes Sep 27, 2025
11bb687
Added tests
RaulMoldes Sep 27, 2025
8275602
Added another test
RaulMoldes Sep 27, 2025
39944b2
Cargo xtask validate passed
RaulMoldes Sep 28, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions crates/cubecl-cpp/src/hip/dialect.rs
Original file line number Diff line number Diff line change
Expand Up @@ -409,7 +409,7 @@ impl<M: DialectWmmaCompiler<Self>> DialectInstructions<Self> for HipDialect<M> {
_rhs: impl Display,
_item: Item<Self>,
) -> std::fmt::Result {
unimplemented!("No native instrution exists, Should be replaced in a preprocessor");
unimplemented!("No native instruction exists, Should be replaced in a preprocessor");
}

fn compile_saturating_sub(
Expand All @@ -418,7 +418,7 @@ impl<M: DialectWmmaCompiler<Self>> DialectInstructions<Self> for HipDialect<M> {
_rhs: impl Display,
_item: Item<Self>,
) -> std::fmt::Result {
unimplemented!("No native instrution exists, Should be replaced in a preprocessor");
unimplemented!("No native instruction exists, Should be replaced in a preprocessor");
}

// others
Expand Down
2 changes: 2 additions & 0 deletions crates/cubecl-cpu/src/runtime.rs
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,8 @@ fn create_client(options: RuntimeOptions) -> ComputeClient<Server, Channel> {
max_page_size: max_shared_memory_size as u64,
alignment: ALIGNMENT,
data_transfer_async: false,
virtual_memory_supported: false,
min_granularity: 0,
};

let memory_management =
Expand Down
2 changes: 2 additions & 0 deletions crates/cubecl-cuda/src/compute/storage/cpu.rs
Original file line number Diff line number Diff line change
Expand Up @@ -107,3 +107,5 @@ impl ComputeStorage for PinnedMemoryStorage {
}
}
}

impl cubecl_runtime::storage::VirtualStorage for PinnedMemoryStorage {}
Loading
Loading