Skip to content

Commit

Permalink
spooky
Browse files Browse the repository at this point in the history
  • Loading branch information
scsmithr committed Jan 5, 2025
1 parent f5d4920 commit edb8875
Show file tree
Hide file tree
Showing 54 changed files with 108 additions and 73 deletions.
10 changes: 5 additions & 5 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion crates/rayexec_execution/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ rayexec_parser = { path = "../rayexec_parser" }
# rayexec_bullet = { path = "../rayexec_bullet" }
rayexec_io = { path = "../rayexec_io" }
fmtutil = { path = "../fmtutil" }
iterutil = { path = "../iterutil" }
stdutil = { path = "../stdutil" }
# stackutil = { path = "../stackutil" } TODO: psm hash issues when compiling to wasm on macos

ahash = { workspace = true }
Expand Down
2 changes: 1 addition & 1 deletion crates/rayexec_execution/src/arrays/array/exp.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
use half::f16;
use iterutil::{IntoExactSizeIterator, TryFromExactSizeIterator};
use rayexec_error::{not_implemented, RayexecError, Result};
use stdutil::iter::{IntoExactSizeIterator, TryFromExactSizeIterator};

use super::array_data::ArrayData;
use super::flat::FlatArrayView;
Expand Down
4 changes: 2 additions & 2 deletions crates/rayexec_execution/src/arrays/batch_exp.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use iterutil::IntoExactSizeIterator;
use rayexec_error::{RayexecError, Result};
use stdutil::iter::IntoExactSizeIterator;

use super::array::exp::Array;
use super::array::selection::Selection;
Expand Down Expand Up @@ -139,7 +139,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;

Expand Down
4 changes: 2 additions & 2 deletions crates/rayexec_execution/src/arrays/compute/cast/array.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
use std::ops::Mul;

use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use num::{CheckedDiv, CheckedMul, Float, NumCast, PrimInt, ToPrimitive};
use rayexec_error::{RayexecError, Result};

Expand Down Expand Up @@ -840,7 +840,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::datatype::DecimalTypeMeta;
Expand Down
2 changes: 1 addition & 1 deletion crates/rayexec_execution/src/arrays/compute/date.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use chrono::{DateTime, Datelike, NaiveDate, Timelike, Utc};
use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use rayexec_error::{not_implemented, RayexecError, Result};

use crate::arrays::array::exp::Array;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use iterutil::IntoExactSizeIterator;
use rayexec_error::Result;
use stdutil::iter::IntoExactSizeIterator;

use super::AggregateState;
use crate::arrays::array::exp::Array;
Expand Down Expand Up @@ -62,7 +62,7 @@ impl BinaryNonNullUpdater {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::physical_type::{AddressableMut, PhysicalI32};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@ impl StateCombiner {
) -> Result<()>
where
State: AggregateState<Input, Output>,
Output: ?Sized,
{
for (from, to) in mapping {
let consume = &mut consume[from];
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use iterutil::IntoExactSizeIterator;
use rayexec_error::Result;
use stdutil::iter::IntoExactSizeIterator;

use super::AggregateState;
use crate::arrays::array::exp::Array;
Expand Down Expand Up @@ -94,7 +94,7 @@ impl UnaryNonNullUpdater {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use rayexec_error::Result;

use crate::arrays::array::exp::Array;
Expand Down Expand Up @@ -144,7 +144,7 @@ impl BinaryExecutor {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::array::validity::Validity;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use rayexec_error::{RayexecError, Result};

use crate::arrays::array::exp::Array;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use rayexec_error::Result;

use crate::arrays::array::exp::Array;
Expand Down Expand Up @@ -189,7 +189,7 @@ impl TernaryExecutor {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use iterutil::IntoExactSizeIterator;
use rayexec_error::Result;
use stdutil::iter::IntoExactSizeIterator;

use crate::arrays::array::exp::Array;
use crate::arrays::array::flat::FlatArrayView;
Expand Down Expand Up @@ -179,7 +179,7 @@ impl UnaryExecutor {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::array::validity::Validity;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use rayexec_error::Result;

use crate::arrays::array::exp::Array;
Expand Down Expand Up @@ -149,7 +149,7 @@ impl UniformExecutor {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
4 changes: 2 additions & 2 deletions crates/rayexec_execution/src/arrays/testutil.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
use std::collections::BTreeMap;
use std::fmt::Debug;

use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;

use super::array::exp::Array;
use super::batch_exp::Batch;
Expand Down Expand Up @@ -201,7 +201,7 @@ pub fn assert_batches_eq(batch1: &Batch, batch2: &Batch) {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
2 changes: 1 addition & 1 deletion crates/rayexec_execution/src/expr/physical/case_expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@ impl fmt::Display for PhysicalCaseExpr {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::datatype::DataType;
Expand Down
2 changes: 1 addition & 1 deletion crates/rayexec_execution/src/expr/physical/cast_expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ impl DatabaseProtoConv for PhysicalCastExpr {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
2 changes: 1 addition & 1 deletion crates/rayexec_execution/src/expr/physical/column_expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ impl DatabaseProtoConv for PhysicalColumnExpr {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::datatype::DataType;
Expand Down
2 changes: 1 addition & 1 deletion crates/rayexec_execution/src/expr/physical/literal_expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ impl DatabaseProtoConv for PhysicalLiteralExpr {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::datatype::DataType;
Expand Down
26 changes: 17 additions & 9 deletions crates/rayexec_execution/src/functions/aggregate/states.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,9 @@ use std::any::Any;
use std::fmt::Debug;
use std::marker::PhantomData;

use iterutil::IntoExactSizeIterator;
use rayexec_error::{RayexecError, Result};
use stdutil::iter::IntoExactSizeIterator;
use stdutil::marker::PhantomCovariant;

use super::ChunkGroupAddressIter;
use crate::arrays::array::exp::Array;
Expand Down Expand Up @@ -32,7 +33,14 @@ use crate::arrays::executor_exp::aggregate::{AggregateState, StateCombiner};
use crate::arrays::executor_exp::PutBuffer;
use crate::arrays::storage::{AddressableStorage, PrimitiveStorage};

pub struct TypedAggregateGroupStates<State, Input, Output, StateInit, StateUpdate, StateFinalize> {
pub struct TypedAggregateGroupStates<
State,
Input,
Output: ?Sized,
StateInit,
StateUpdate,
StateFinalize,
> {
/// States being tracked.
states: Vec<State>,

Expand All @@ -47,11 +55,11 @@ pub struct TypedAggregateGroupStates<State, Input, Output, StateInit, StateUpdat
/// How to finalize states.
state_finalize: StateFinalize,

_input: PhantomData<Input>,
_output: PhantomData<Output>,
_input: PhantomCovariant<Input>,
_output: PhantomCovariant<Output>,
}

impl<State, Input, Output, StateInit, StateUpdate, StateFinalize>
impl<State, Input, Output: ?Sized, StateInit, StateUpdate, StateFinalize>
TypedAggregateGroupStates<State, Input, Output, StateInit, StateUpdate, StateFinalize>
{
pub fn new(
Expand All @@ -65,8 +73,8 @@ impl<State, Input, Output, StateInit, StateUpdate, StateFinalize>
state_init,
state_update,
state_finalize,
_input: PhantomData,
_output: PhantomData,
_input: PhantomCovariant::new(),
_output: PhantomCovariant::new(),
}
}
}
Expand All @@ -76,7 +84,7 @@ impl<State, Input, Output, StateInit, StateUpdate, StateFinalize> AggregateGroup
where
State: AggregateState<Input, Output> + Sync + Send + 'static,
Input: Sync + Send,
Output: Sync + Send,
Output: Sync + Send + ?Sized,
StateInit: Fn() -> State + Sync + Send,
StateUpdate: Fn(&[Array], Selection, &[usize], &mut [State]) -> Result<()> + Sync + Send,
StateFinalize: Fn(&mut [State], &mut Array) -> Result<()> + Sync + Send,
Expand Down Expand Up @@ -155,7 +163,7 @@ where
Ok(())
}

impl<State, Input, Output, StateInit, StateUpdate, StateFinalize> fmt::Debug
impl<State, Input, Output: ?Sized, StateInit, StateUpdate, StateFinalize> fmt::Debug
for TypedAggregateGroupStates<State, Input, Output, StateInit, StateUpdate, StateFinalize>
{
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -317,7 +317,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -212,7 +212,7 @@ where

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,7 @@ impl ScalarFunctionImpl for OrImpl {

#[cfg(test)]
mod tests {
use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -854,7 +854,7 @@ where
#[cfg(test)]
mod tests {

use iterutil::TryFromExactSizeIterator;
use stdutil::iter::TryFromExactSizeIterator;

use super::*;
use crate::arrays::buffer::buffer_manager::NopBufferManager;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
use iterutil::IntoExactSizeIterator;
use stdutil::iter::IntoExactSizeIterator;
use rayexec_error::Result;

use crate::arrays::array::exp::Array;
Expand Down
Loading

0 comments on commit edb8875

Please sign in to comment.