Home
last modified time | relevance | path

Searched refs:KernelArgType (Results 1 – 8 of 8) sorted by relevance

/external/mesa3d/src/gallium/frontends/rusticl/api/
Dkernel.rs367 KernelArgType::MemLocal => { in set_kernel_arg()
372 KernelArgType::MemGlobal in set_kernel_arg()
373 | KernelArgType::MemConstant in set_kernel_arg()
374 | KernelArgType::Image in set_kernel_arg()
375 | KernelArgType::RWImage in set_kernel_arg()
376 | KernelArgType::Texture => { in set_kernel_arg()
382 KernelArgType::Sampler => { in set_kernel_arg()
388 KernelArgType::Constant(size) => { in set_kernel_arg()
399 KernelArgType::MemLocal => { in set_kernel_arg()
406 KernelArgType::Constant(_) | KernelArgType::Sampler => { in set_kernel_arg()
[all …]
/external/mesa3d/src/gallium/frontends/rusticl/core/
Dkernel.rs60 pub enum KernelArgType { enum
71 impl KernelArgType { implementation
78 KernelArgType::Constant(size) in deserialize()
80 1 => KernelArgType::Image, in deserialize()
81 2 => KernelArgType::RWImage, in deserialize()
82 3 => KernelArgType::Sampler, in deserialize()
83 4 => KernelArgType::Texture, in deserialize()
84 5 => KernelArgType::MemGlobal, in deserialize()
85 6 => KernelArgType::MemConstant, in deserialize()
86 7 => KernelArgType::MemLocal, in deserialize()
[all …]
/external/pytorch/torch/_inductor/codegen/
Dtriton_utils.py12 from .common import KernelArgType, SizeArg, TensorArg, WorkspaceArg
26 def signature_of(arg: KernelArgType, *, size_dtype: str) -> str: argument
68 signature: List[KernelArgType], argument
109 args: List[KernelArgType], argument
116 def is_aligned(x: KernelArgType, alignment: int, include_tensor: bool) -> bool: argument
Dwrapper.py1267 from .common import KernelArgType, SizeArg, TensorArg
1269 signature: List[KernelArgType] = []
Dcommon.py94 KernelArgType = Union[WorkspaceArg, TensorArg, SizeArg] variable
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConvolution.h292 template<typename Indices, typename InputArgType, typename KernelArgType, typename Device>
293 struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelArgType>, Device>
295 typedef TensorConvolutionOp<Indices, InputArgType, KernelArgType> XprType;
310 …nsorEvaluator<InputArgType, Device>::IsAligned) & int(TensorEvaluator<KernelArgType, Device>::IsAl…
311 …rEvaluator<InputArgType, Device>::PacketAccess) & int(TensorEvaluator<KernelArgType, Device>::Pack…
326 …or<InputArgType, Device>::Layout) == static_cast<int>(TensorEvaluator<KernelArgType, Device>::Layo…
329 …const typename TensorEvaluator<KernelArgType, Device>::Dimensions& kernel_dims = m_kernelImpl.dime…
535 typedef TensorEvalToOp<const KernelArgType> EvalTo;
537 const bool Vectorize = internal::IsVectorizable<Device, KernelArgType>::value;
551 TensorEvaluator<KernelArgType, Device> m_kernelImpl;
[all …]
DTensorConvolutionSycl.h274 template <typename Indices, typename InputArgType, typename KernelArgType>
275 struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelArgType>, Eigen::Sycl…
276 typedef TensorConvolutionOp<Indices, InputArgType, KernelArgType> XprType;
283 typedef typename TensorEvaluator<KernelArgType, Eigen::SyclDevice>::Dimensions KernelDimensions;
295 TensorEvaluator<KernelArgType, Eigen::SyclDevice>::IsAligned,
318 … static_cast<int>(TensorEvaluator<KernelArgType, Eigen::SyclDevice>::Layout)),
322 const typename TensorEvaluator<KernelArgType, Eigen::SyclDevice>::Dimensions &kernel_dims =
378 typedef TensorEvalToOp<const KernelArgType> EvalTo;
380 const bool PacketAccess = internal::IsVectorizable<Eigen::SyclDevice, KernelArgType>::value;
532 KernelArgType m_kernelArg;
[all …]
/external/mesa3d/docs/relnotes/
D24.3.0.rst3604 - rusticl/kernel: move serialization code of KernelArgType
3606 - rusticl/kernel: add KernelArgType::is_opaque