#![allow(unused_imports)]
use crate::{vk::Result as VkResult, vk::*, *};
use core::ffi::{CStr, c_char, c_int, c_void};
use core::mem::transmute;
use core::ptr;
pub const EXTENSION_NAME: &CStr = c"VK_NV_cooperative_matrix";
pub(super) mod defs {
#![allow(non_camel_case_types, unused_imports)]
use crate::{vk::*, *};
use core::ffi::{CStr, c_char, c_int, c_void};
use core::fmt;
use core::marker::PhantomData;
use core::ptr;
pub type ScopeNV = ScopeKHR;
pub type ComponentTypeNV = ComponentTypeKHR;
#[repr(C)]
#[derive(Copy, Clone)]
#[must_use]
pub struct PhysicalDeviceCooperativeMatrixFeaturesNV<'a> {
pub s_type: StructureType,
pub p_next: *mut c_void,
pub cooperative_matrix: Bool32,
pub cooperative_matrix_robust_buffer_access: Bool32,
pub _marker: PhantomData<&'a ()>,
}
#[cfg(feature = "debug")]
impl fmt::Debug for PhysicalDeviceCooperativeMatrixFeaturesNV<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("PhysicalDeviceCooperativeMatrixFeaturesNV")
.field("s_type", &self.s_type)
.field("p_next", &self.p_next)
.field("cooperative_matrix", &self.cooperative_matrix)
.field(
"cooperative_matrix_robust_buffer_access",
&self.cooperative_matrix_robust_buffer_access,
)
.finish()
}
}
unsafe impl<'a> TaggedStructure<'a> for PhysicalDeviceCooperativeMatrixFeaturesNV<'a> {
const STRUCTURE_TYPE: StructureType =
StructureType::PHYSICAL_DEVICE_COOPERATIVE_MATRIX_FEATURES_NV;
}
unsafe impl Extends<PhysicalDeviceFeatures2<'_>> for PhysicalDeviceCooperativeMatrixFeaturesNV<'_> {}
unsafe impl Extends<DeviceCreateInfo<'_>> for PhysicalDeviceCooperativeMatrixFeaturesNV<'_> {}
impl Default for PhysicalDeviceCooperativeMatrixFeaturesNV<'_> {
fn default() -> Self {
Self {
s_type: Self::STRUCTURE_TYPE,
p_next: ptr::null_mut(),
cooperative_matrix: Default::default(),
cooperative_matrix_robust_buffer_access: Default::default(),
_marker: PhantomData,
}
}
}
impl<'a> PhysicalDeviceCooperativeMatrixFeaturesNV<'a> {
#[inline]
pub fn cooperative_matrix(mut self, cooperative_matrix: bool) -> Self {
self.cooperative_matrix = cooperative_matrix.into();
self
}
#[inline]
pub fn cooperative_matrix_robust_buffer_access(
mut self,
cooperative_matrix_robust_buffer_access: bool,
) -> Self {
self.cooperative_matrix_robust_buffer_access =
cooperative_matrix_robust_buffer_access.into();
self
}
}
#[repr(C)]
#[derive(Copy, Clone)]
#[must_use]
pub struct PhysicalDeviceCooperativeMatrixPropertiesNV<'a> {
pub s_type: StructureType,
pub p_next: *mut c_void,
pub cooperative_matrix_supported_stages: ShaderStageFlags,
pub _marker: PhantomData<&'a ()>,
}
#[cfg(feature = "debug")]
impl fmt::Debug for PhysicalDeviceCooperativeMatrixPropertiesNV<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("PhysicalDeviceCooperativeMatrixPropertiesNV")
.field("s_type", &self.s_type)
.field("p_next", &self.p_next)
.field(
"cooperative_matrix_supported_stages",
&self.cooperative_matrix_supported_stages,
)
.finish()
}
}
unsafe impl<'a> TaggedStructure<'a> for PhysicalDeviceCooperativeMatrixPropertiesNV<'a> {
const STRUCTURE_TYPE: StructureType =
StructureType::PHYSICAL_DEVICE_COOPERATIVE_MATRIX_PROPERTIES_NV;
}
unsafe impl Extends<PhysicalDeviceProperties2<'_>>
for PhysicalDeviceCooperativeMatrixPropertiesNV<'_>
{
}
impl Default for PhysicalDeviceCooperativeMatrixPropertiesNV<'_> {
fn default() -> Self {
Self {
s_type: Self::STRUCTURE_TYPE,
p_next: ptr::null_mut(),
cooperative_matrix_supported_stages: Default::default(),
_marker: PhantomData,
}
}
}
impl<'a> PhysicalDeviceCooperativeMatrixPropertiesNV<'a> {
#[inline]
pub fn cooperative_matrix_supported_stages(
mut self,
cooperative_matrix_supported_stages: ShaderStageFlags,
) -> Self {
self.cooperative_matrix_supported_stages = cooperative_matrix_supported_stages;
self
}
}
#[repr(C)]
#[derive(Copy, Clone)]
#[must_use]
pub struct CooperativeMatrixPropertiesNV<'a> {
pub s_type: StructureType,
pub p_next: *mut c_void,
pub m_size: u32,
pub n_size: u32,
pub k_size: u32,
pub a_type: ComponentTypeNV,
pub b_type: ComponentTypeNV,
pub c_type: ComponentTypeNV,
pub d_type: ComponentTypeNV,
pub scope: ScopeNV,
pub _marker: PhantomData<&'a ()>,
}
#[cfg(feature = "debug")]
impl fmt::Debug for CooperativeMatrixPropertiesNV<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("CooperativeMatrixPropertiesNV")
.field("s_type", &self.s_type)
.field("p_next", &self.p_next)
.field("m_size", &self.m_size)
.field("n_size", &self.n_size)
.field("k_size", &self.k_size)
.field("a_type", &self.a_type)
.field("b_type", &self.b_type)
.field("c_type", &self.c_type)
.field("d_type", &self.d_type)
.field("scope", &self.scope)
.finish()
}
}
unsafe impl<'a> TaggedStructure<'a> for CooperativeMatrixPropertiesNV<'a> {
const STRUCTURE_TYPE: StructureType = StructureType::COOPERATIVE_MATRIX_PROPERTIES_NV;
}
impl Default for CooperativeMatrixPropertiesNV<'_> {
fn default() -> Self {
Self {
s_type: Self::STRUCTURE_TYPE,
p_next: ptr::null_mut(),
m_size: Default::default(),
n_size: Default::default(),
k_size: Default::default(),
a_type: Default::default(),
b_type: Default::default(),
c_type: Default::default(),
d_type: Default::default(),
scope: Default::default(),
_marker: PhantomData,
}
}
}
impl<'a> CooperativeMatrixPropertiesNV<'a> {
#[inline]
pub fn m_size(mut self, m_size: u32) -> Self {
self.m_size = m_size;
self
}
#[inline]
pub fn n_size(mut self, n_size: u32) -> Self {
self.n_size = n_size;
self
}
#[inline]
pub fn k_size(mut self, k_size: u32) -> Self {
self.k_size = k_size;
self
}
#[inline]
pub fn a_type(mut self, a_type: ComponentTypeNV) -> Self {
self.a_type = a_type;
self
}
#[inline]
pub fn b_type(mut self, b_type: ComponentTypeNV) -> Self {
self.b_type = b_type;
self
}
#[inline]
pub fn c_type(mut self, c_type: ComponentTypeNV) -> Self {
self.c_type = c_type;
self
}
#[inline]
pub fn d_type(mut self, d_type: ComponentTypeNV) -> Self {
self.d_type = d_type;
self
}
#[inline]
pub fn scope(mut self, scope: ScopeNV) -> Self {
self.scope = scope;
self
}
}
pub type PFN_vkGetPhysicalDeviceCooperativeMatrixPropertiesNV =
unsafe extern "system" fn(
physical_device: PhysicalDevice,
p_property_count: *mut u32,
p_properties: *mut CooperativeMatrixPropertiesNV<'_>,
) -> vk::Result;
}
#[cfg(feature = "ffi")]
pub(super) mod ffi {
#![allow(non_camel_case_types)]
use super::defs::*;
pub type VkPhysicalDeviceCooperativeMatrixFeaturesNV =
PhysicalDeviceCooperativeMatrixFeaturesNV<'static>;
pub type VkPhysicalDeviceCooperativeMatrixPropertiesNV =
PhysicalDeviceCooperativeMatrixPropertiesNV<'static>;
pub type VkCooperativeMatrixPropertiesNV = CooperativeMatrixPropertiesNV<'static>;
pub type VkScopeNV = ScopeNV;
pub type VkComponentTypeNV = ComponentTypeNV;
impl PhysicalDeviceCooperativeMatrixFeaturesNV<'_> {
#[inline]
pub unsafe fn drop_lifetime_for_ffi(&self) -> &VkPhysicalDeviceCooperativeMatrixFeaturesNV {
unsafe { core::mem::transmute(self) }
}
}
impl PhysicalDeviceCooperativeMatrixPropertiesNV<'_> {
#[inline]
pub unsafe fn drop_lifetime_for_ffi(
&self,
) -> &VkPhysicalDeviceCooperativeMatrixPropertiesNV {
unsafe { core::mem::transmute(self) }
}
}
impl CooperativeMatrixPropertiesNV<'_> {
#[inline]
pub unsafe fn drop_lifetime_for_ffi(&self) -> &VkCooperativeMatrixPropertiesNV {
unsafe { core::mem::transmute(self) }
}
}
}
pub struct InstanceFn {
get_physical_device_cooperative_matrix_properties:
PFN_vkGetPhysicalDeviceCooperativeMatrixPropertiesNV,
}
impl LoadInstanceFn for InstanceFn {
unsafe fn load_with(
load: impl Fn(&CStr) -> Option<PFN_vkVoidFunction>,
) -> core::result::Result<Self, MissingEntryPointError> {
unsafe {
Ok(Self {
get_physical_device_cooperative_matrix_properties: transmute(
load(c"vkGetPhysicalDeviceCooperativeMatrixPropertiesNV")
.ok_or(MissingEntryPointError)?,
),
})
}
}
}
impl InstanceFn {
#[inline]
pub unsafe fn get_physical_device_cooperative_matrix_properties<'a>(
&self,
physical_device: PhysicalDevice,
mut properties: impl EnumerateInto<CooperativeMatrixPropertiesNV<'a>>,
) -> crate::Result<()> {
unsafe {
let call = |property_count, properties| {
let result = (self.get_physical_device_cooperative_matrix_properties)(
physical_device,
property_count,
properties as _,
);
match result {
VkResult::SUCCESS => Ok(()),
VkResult::INCOMPLETE => Ok(()),
err => Err(err),
}
};
let mut len = 0;
call(&mut len, std::ptr::null_mut())?;
let capacity = len.try_into().expect("failed to convert `N` to usize");
let properties_buf = properties.reserve(capacity);
len = properties_buf.len().try_into().unwrap();
let result = call(&mut len, properties_buf.as_mut_ptr() as *mut _)?;
properties.set_len(len.try_into().unwrap());
Ok(result)
}
}
}