pub struct GroupedProblem<'a, T: Element> {
pub m: i32,
pub n: i32,
pub k: i32,
pub a: MatrixRef<'a, T>,
pub b: MatrixRef<'a, T>,
pub c: Option<MatrixRef<'a, T>>,
pub d: MatrixMut<'a, T>,
pub alpha: T::Scalar,
pub beta: T::Scalar,
}Expand description
One per-group entry for a grouped GEMM launch.
Each group has its own shape and pointers; CUTLASS dispatches them in
a single kernel invocation. Passed as a slice to
GroupedGemmPlan::prepare, which
returns a PreparedGroupedGemm whose
run method performs the launch.
Fields§
§m: i32Group M.
n: i32Group N.
k: i32Group K.
a: MatrixRef<'a, T>Left input.
b: MatrixRef<'a, T>Right input.
c: Option<MatrixRef<'a, T>>Optional accumulation source.
d: MatrixMut<'a, T>Output.
alpha: T::Scalar§beta: T::Scalarβ for this group. Forced to 0 internally when c is None.
Trait Implementations§
Auto Trait Implementations§
impl<'a, T> !UnwindSafe for GroupedProblem<'a, T>
impl<'a, T> Freeze for GroupedProblem<'a, T>
impl<'a, T> RefUnwindSafe for GroupedProblem<'a, T>
impl<'a, T> Send for GroupedProblem<'a, T>
impl<'a, T> Sync for GroupedProblem<'a, T>
impl<'a, T> Unpin for GroupedProblem<'a, T>
impl<'a, T> UnsafeUnpin for GroupedProblem<'a, T>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more