Skip to content

Commit 0d2ff1a

Browse files
authored
Resolve tiled slice dimensionality constraint and migrate dsperse (#430)
1 parent a60a609 commit 0d2ff1a

File tree

3 files changed

+4
-11
lines changed

3 files changed

+4
-11
lines changed

Cargo.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ futures-util = "0.3"
5050
shellexpand = "3.1.2"
5151
openssl = { version = "0.10", features = ["vendored"] }
5252
flate2 = "1"
53-
dsperse = { git = "https://github.com/inference-labs-inc/dsperse", rev = "8a0c9ddfc8ad7623451549dea784f5a2339a35d3" }
53+
dsperse = { git = "https://github.com/inference-labs-inc/dsperse", rev = "176ed08ee9552ac23f4a6470d1e7b53b67305b08" }
5454
ndarray = { version = "0.17", features = ["serde"] }
5555
zip = { version = "2", default-features = false, features = ["deflate"] }
5656
bittensor-drand = { git = "https://github.com/inference-labs-inc/bittensor-drand.git", rev = "e55ed98", default-features = false }

crates/sn2-validator/src/validator_loop/dslice.rs

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -322,17 +322,10 @@ impl ValidatorLoop {
322322
input_tensor: ndarray::ArrayD<f64>,
323323
run_source: RunSource,
324324
) -> Option<usize> {
325-
let input_4d = match input_tensor.into_dimensionality::<ndarray::Ix4>() {
326-
Ok(arr) => arr,
327-
Err(e) => {
328-
warn!(run_uid = %run_uid, slice = %slice_id, error = %e, "tiled slice requires 4D input");
329-
return None;
330-
}
331-
};
332-
let tiles = match dsperse::pipeline::split_into_tiles(&input_4d, tiling) {
325+
let tiles = match dsperse::pipeline::split_for_tiling(&input_tensor, tiling) {
333326
Ok(t) => t,
334327
Err(e) => {
335-
warn!(run_uid = %run_uid, slice = %slice_id, error = %e, "split_into_tiles failed");
328+
warn!(run_uid = %run_uid, slice = %slice_id, error = %e, "split_for_tiling failed");
336329
return None;
337330
}
338331
};

0 commit comments

Comments
 (0)