Struct KeyedPtxModuleCache

Source

pub struct KeyedPtxModuleCache<K> { /* private fields */ }

Expand description

A per-key family of compiled modules: one CudaModule per value of a kernel-shape parameter that is baked into the source (a row width P, say), each compiled with the shared device-keyed NVRTC options on first use and shared thereafter. The keyed twin of PtxModuleCache.

Struct KeyedPtxModuleCache Copy item path

Implementations§

impl<K: Eq + Hash + Copy + Display> KeyedPtxModuleCache<K>

pub fn new() -> Self

pub fn get_or_compile<S>( &self, ctx: &Arc<CudaContext>, key: K, label: &'static str, source: S, ) -> Result<Arc<CudaModule>, GpuError>where S: FnOnce(K) -> String,

Trait Implementations§

impl<K: Eq + Hash + Copy + Display> Default for KeyedPtxModuleCache<K>

fn default() -> Self

Auto Trait Implementations§

impl<K> !Freeze for KeyedPtxModuleCache<K>

impl<K> RefUnwindSafe for KeyedPtxModuleCache<K>where Mutex<HashMap<K, Arc<CudaModule>>>: RefUnwindSafe,

impl<K> Send for KeyedPtxModuleCache<K>where Mutex<HashMap<K, Arc<CudaModule>>>: Send,

impl<K> Sync for KeyedPtxModuleCache<K>where Mutex<HashMap<K, Arc<CudaModule>>>: Sync,

impl<K> Unpin for KeyedPtxModuleCache<K>where Mutex<HashMap<K, Arc<CudaModule>>>: Unpin,

impl<K> UnsafeUnpin for KeyedPtxModuleCache<K>where Mutex<HashMap<K, Arc<CudaModule>>>: UnsafeUnpin,

impl<K> UnwindSafe for KeyedPtxModuleCache<K>where Mutex<HashMap<K, Arc<CudaModule>>>: UnwindSafe,

Blanket Implementations§

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

fn type_id(&self) -> TypeId

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

fn borrow(&self) -> &T

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

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

impl<T> ByRef<T> for T

fn by_ref(&self) -> &T

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DTwhere ST: ?Sized, DT: ?Sized,

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DTwhere ST: ?Sized, DT: ?Sized,

impl<T> DistributionExt for Twhere T: ?Sized,

fn rand<T>(&self, rng: &mut (impl Rng + ?Sized)) -> Twhere Self: Distribution<T>,

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Imply<T> for Uwhere T: ?Sized, U: ?Sized,

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

fn into(self) -> U

impl<T> IntoEither for T

fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘwhere F: FnOnce(&Self) -> bool,

impl<T> Pointable for T

const ALIGN: usize

type Init = T

unsafe fn init(init: <T as Pointable>::Init) -> usize

unsafe fn deref<'a>(ptr: usize) -> &'a T

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

unsafe fn drop(ptr: usize)

impl<T> Read<Exclusive, BecauseExclusive> for Twhere T: ?Sized,

impl<T> Same for T

type Output = T

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

type Error = !

fn try_from(value: U) -> Result<T, !>

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

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

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

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

fn vzip(self) -> V

Struct KeyedPtxModuleCache

pub fn get_or_compile<S>( &self, ctx: &Arc<CudaContext>, key: K, label: &'static str, source: S, ) -> Result<Arc<CudaModule>, GpuError>
where S: FnOnce(K) -> String,

impl<K> RefUnwindSafe for KeyedPtxModuleCache<K>
where Mutex<HashMap<K, Arc<CudaModule>>>: RefUnwindSafe,

impl<K> Send for KeyedPtxModuleCache<K>
where Mutex<HashMap<K, Arc<CudaModule>>>: Send,

impl<K> Sync for KeyedPtxModuleCache<K>
where Mutex<HashMap<K, Arc<CudaModule>>>: Sync,

impl<K> Unpin for KeyedPtxModuleCache<K>
where Mutex<HashMap<K, Arc<CudaModule>>>: Unpin,

impl<K> UnsafeUnpin for KeyedPtxModuleCache<K>
where Mutex<HashMap<K, Arc<CudaModule>>>: UnsafeUnpin,

impl<K> UnwindSafe for KeyedPtxModuleCache<K>
where Mutex<HashMap<K, Arc<CudaModule>>>: UnwindSafe,

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

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

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

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

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

fn rand<T>(&self, rng: &mut (impl Rng + ?Sized)) -> T
where Self: Distribution<T>,

impl<T, U> Imply<T> for U
where T: ?Sized, U: ?Sized,

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

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
where F: FnOnce(&Self) -> bool,

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

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

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

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