Skip to content

Commit 19e1256

Browse files
0ax10ax1
committed
style: rustfmt (nightly)
Signed-off-by: Alexander Droste <alexander.droste@protonmail.com>
1 parent b9ac775 commit 19e1256

1 file changed

Lines changed: 60 additions & 12 deletions

File tree

  • vortex-cuda/src/dynamic_dispatch

‎vortex-cuda/src/dynamic_dispatch/mod.rs‎

Lines changed: 60 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2137,7 +2137,11 @@ mod tests {
21372137
.collect();
21382138

21392139
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2140-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2140+
let bp = BitPacked::encode(
2141+
&prim.into_array(),
2142+
bit_width,
2143+
&mut LEGACY_SESSION.create_execution_ctx(),
2144+
)?;
21412145
assert!(bp.patches().is_some(), "expected patches");
21422146

21432147
// Slice crossing chunk boundaries.
@@ -2174,7 +2178,11 @@ mod tests {
21742178
let all_values: Vec<u32> = residuals.iter().map(|&v| v + reference).collect();
21752179

21762180
let prim = PrimitiveArray::new(Buffer::from(residuals), NonNullable);
2177-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2181+
let bp = BitPacked::encode(
2182+
&prim.into_array(),
2183+
bit_width,
2184+
&mut LEGACY_SESSION.create_execution_ctx(),
2185+
)?;
21782186
assert!(bp.patches().is_some(), "expected patches");
21792187
let for_arr = FoR::try_new(bp.into_array(), Scalar::from(reference))?;
21802188

@@ -2319,7 +2327,11 @@ mod tests {
23192327
.collect();
23202328

23212329
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2322-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2330+
let bp = BitPacked::encode(
2331+
&prim.into_array(),
2332+
bit_width,
2333+
&mut LEGACY_SESSION.create_execution_ctx(),
2334+
)?;
23232335
assert!(bp.patches().is_some(), "expected patches");
23242336

23252337
let cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;
@@ -2348,7 +2360,11 @@ mod tests {
23482360
let expected: Vec<u32> = residuals.iter().map(|&v| v + reference).collect();
23492361

23502362
let prim = PrimitiveArray::new(Buffer::from(residuals), NonNullable);
2351-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2363+
let bp = BitPacked::encode(
2364+
&prim.into_array(),
2365+
bit_width,
2366+
&mut LEGACY_SESSION.create_execution_ctx(),
2367+
)?;
23522368
assert!(bp.patches().is_some(), "expected patches");
23532369
let for_arr = FoR::try_new(bp.into_array(), Scalar::from(reference))?;
23542370

@@ -2406,7 +2422,11 @@ mod tests {
24062422
.collect();
24072423

24082424
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2409-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2425+
let bp = BitPacked::encode(
2426+
&prim.into_array(),
2427+
bit_width,
2428+
&mut LEGACY_SESSION.create_execution_ctx(),
2429+
)?;
24102430
assert!(bp.patches().is_some(), "expected patches");
24112431

24122432
let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;
@@ -2435,7 +2455,11 @@ mod tests {
24352455
.collect();
24362456

24372457
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2438-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2458+
let bp = BitPacked::encode(
2459+
&prim.into_array(),
2460+
bit_width,
2461+
&mut LEGACY_SESSION.create_execution_ctx(),
2462+
)?;
24392463
assert!(bp.patches().is_some(), "expected patches");
24402464

24412465
let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;
@@ -2464,7 +2488,11 @@ mod tests {
24642488
.collect();
24652489

24662490
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2467-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2491+
let bp = BitPacked::encode(
2492+
&prim.into_array(),
2493+
bit_width,
2494+
&mut LEGACY_SESSION.create_execution_ctx(),
2495+
)?;
24682496
assert!(bp.patches().is_some(), "expected patches");
24692497

24702498
let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;
@@ -2496,7 +2524,11 @@ mod tests {
24962524
let expected: Vec<u32> = codes.iter().map(|&c| dict_values[c as usize]).collect();
24972525

24982526
let codes_prim = PrimitiveArray::new(Buffer::from(codes), NonNullable);
2499-
let codes_bp = BitPacked::encode(&codes_prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2527+
let codes_bp = BitPacked::encode(
2528+
&codes_prim.into_array(),
2529+
bit_width,
2530+
&mut LEGACY_SESSION.create_execution_ctx(),
2531+
)?;
25002532
assert!(codes_bp.patches().is_some(), "expected patches on codes");
25012533

25022534
let values_prim = PrimitiveArray::new(Buffer::from(dict_values), NonNullable);
@@ -2524,7 +2556,11 @@ mod tests {
25242556
values[2048] = 4000; // start of chunk 2
25252557

25262558
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2527-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2559+
let bp = BitPacked::encode(
2560+
&prim.into_array(),
2561+
bit_width,
2562+
&mut LEGACY_SESSION.create_execution_ctx(),
2563+
)?;
25282564
assert!(bp.patches().is_some(), "expected patches");
25292565

25302566
let cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;
@@ -2552,7 +2588,11 @@ mod tests {
25522588
.collect();
25532589

25542590
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2555-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2591+
let bp = BitPacked::encode(
2592+
&prim.into_array(),
2593+
bit_width,
2594+
&mut LEGACY_SESSION.create_execution_ctx(),
2595+
)?;
25562596
assert!(bp.patches().is_some(), "expected patches");
25572597

25582598
let cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;
@@ -2589,7 +2629,11 @@ mod tests {
25892629
let prim = PrimitiveArray::from_option_iter(values.iter().copied());
25902630
let cpu = crate::canonicalize_cpu(prim.clone())?.into_array();
25912631

2592-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2632+
let bp = BitPacked::encode(
2633+
&prim.into_array(),
2634+
bit_width,
2635+
&mut LEGACY_SESSION.create_execution_ctx(),
2636+
)?;
25932637
assert!(bp.patches().is_some(), "expected patches");
25942638

25952639
let gpu = try_gpu_dispatch(&bp.into_array(), &mut cuda_ctx)
@@ -2612,7 +2656,11 @@ mod tests {
26122656
let values: Vec<u32> = (0..len).map(|i| (i as u32) + 2).collect();
26132657

26142658
let prim = PrimitiveArray::new(Buffer::from(values.clone()), NonNullable);
2615-
let bp = BitPacked::encode(&prim.into_array(), bit_width, &mut LEGACY_SESSION.create_execution_ctx())?;
2659+
let bp = BitPacked::encode(
2660+
&prim.into_array(),
2661+
bit_width,
2662+
&mut LEGACY_SESSION.create_execution_ctx(),
2663+
)?;
26162664
assert!(bp.patches().is_some(), "expected patches");
26172665

26182666
let cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())?;

0 commit comments

Comments
 (0)