Fix UB in argument.rs

gfx-rs · Sep 1, 2022 · fb46081 · fb46081
1 parent bd27144
commit fb46081
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/argument.rs b/src/argument.rs
@@ -152,11 +152,11 @@ impl StructMemberRef {
         unsafe { msg_send![self, dataType] }
     }
 
-    pub fn struct_type(&self) -> MTLStructType {
+    pub fn struct_type(&self) -> &StructTypeRef {
         unsafe { msg_send![self, structType] }
     }
 
-    pub fn array_type(&self) -> MTLArrayType {
+    pub fn array_type(&self) -> &ArrayTypeRef {
         unsafe { msg_send![self, arrayType] }
     }
 }
@@ -220,11 +220,11 @@ impl ArrayTypeRef {
         unsafe { msg_send![self, elementType] }
     }
 
-    pub fn element_struct_type(&self) -> MTLStructType {
+    pub fn element_struct_type(&self) -> &StructTypeRef {
         unsafe { msg_send![self, elementStructType] }
     }
 
-    pub fn element_array_type(&self) -> MTLArrayType {
+    pub fn element_array_type(&self) -> &ArrayTypeRef {
         unsafe { msg_send![self, elementArrayType] }
     }
 }