Fp8ActQuant

Struct Fp8ActQuant 

Source
pub struct Fp8ActQuant {
    pub shared: KernelHandle,
    pub hopper: KernelHandle,
}
Expand description

The FP8 activation quantizer a layer will launch: the shared kernel, plus the Hopper twin when this target has one.

Fields§

§shared: KernelHandle

per_token_group_quant_fp8, present on every target.

§hopper: KernelHandle

per_token_group_quant_fp8_hopper, or KernelHandle(0) off Hopper.

Implementations§

Source§

impl Fp8ActQuant

Source

pub fn resolve(gpu: &dyn GpuBackend) -> Self

Probe both. try_kernel for each: the shared one is optional on targets with no FP8 path at all (the callers already gate on Self::available), and the twin is optional everywhere but Hopper.

Source

pub fn shared_only(shared: KernelHandle) -> Self

A quantizer with only the shared kernel — the state every non-Hopper target resolves to, and the one unit tests construct.

Source

pub fn available(&self) -> bool

Is there a quantizer to launch at all? The replacement for the handle.0 != 0 test every W8A8 selector used to spell inline.

EITHER handle, not Self::pick’s: the pick is width-dependent, and a caller asking “is there an FP8 path on this target” is not asking about one launch’s M.

Source

pub fn twin_present(&self) -> bool

Is the twin in this image? PRESENCE, which is a property of the build — not “will the next launch use it”, which is Fp8QuantPick::twin.

Source

pub fn pick(&self, m: u32, k: u32) -> Fp8QuantPick

Which kernel this (m, k) launches, on which grid, and why not the other one — the process’s resolved lever and the compiled target’s SM count. See Self::pick_with for the pure form.

Source

pub fn pick_with( &self, requested: bool, m: u32, k: u32, sm_count: u32, ) -> Fp8QuantPick

Self::pick over an explicit lever and SM count — pure, so every (M, K) the attribution prices is gradeable from a CPU test.

Source

pub fn kernel(&self, m: u32, k: u32) -> KernelHandle

The handle this (m, k) launches.

Source

pub fn grid(&self, m: u32, k: u32) -> [u32; 3]

The grid for (m, k), matching Self::kernel — from the SAME pick, so a twin handle can never reach the parent’s grid.

Trait Implementations§

Source§

impl Clone for Fp8ActQuant

Source§

fn clone(&self) -> Fp8ActQuant

Returns a duplicate of the value. Read more
1.0.0 · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Debug for Fp8ActQuant

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl Default for Fp8ActQuant

Source§

fn default() -> Self

No quantizer at all — what a target with no FP8 path resolves, and the negative every W8A8 selector is tested against.

Source§

impl Copy for Fp8ActQuant

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

§

impl<T> Instrument for T

§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more
§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

§

fn vzip(self) -> V

§

impl<T> WithSubscriber for T

§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more
§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more