@@ -229,14 +229,14 @@ impl InstanceOp {
229229 builder. build_with ( |builder, state| {
230230 let num_inputs = inputs. into_iter ( ) . map ( |v| state. add_operand ( v) ) . count ( ) ;
231231 let num_outputs = outputs. into_iter ( ) . map ( |v| state. add_operand ( v) ) . count ( ) ;
232- let vector_attr = Attribute :: from_raw ( unsafe {
233- mlirDenseElementsAttrInt32Get (
234- mlirVectorTypeGet ( 1 , [ 2 ] . as_ptr ( ) , get_integer_type ( builder. cx , 32 ) . raw ( ) ) ,
232+ let dense_array_attr = Attribute :: from_raw ( unsafe {
233+ mlirDenseI32ArrayGet (
234+ builder. cx . raw ( ) ,
235235 2 ,
236236 [ num_inputs as _ , num_outputs as _ ] . as_ptr ( ) ,
237237 )
238238 } ) ;
239- state. add_attribute ( "operand_segment_sizes" , vector_attr ) ;
239+ state. add_attribute ( "operand_segment_sizes" , dense_array_attr ) ;
240240 state. add_attribute ( "name" , get_string_attr ( builder. cx , name) ) ;
241241 state. add_attribute ( "callee" , get_flat_symbol_ref_attr ( builder. cx , module) ) ;
242242 } )
@@ -383,17 +383,14 @@ impl WaitOp {
383383 if let Some ( time) = time {
384384 state. add_operand ( time) ;
385385 }
386- let vector_ty = unsafe {
387- mlirVectorTypeGet ( 1 , [ 3 ] . as_ptr ( ) , get_integer_type ( builder. cx , 32 ) . raw ( ) )
388- } ;
389- let vector_attr = Attribute :: from_raw ( unsafe {
390- mlirDenseElementsAttrInt32Get (
391- vector_ty,
386+ let dense_array_attr = Attribute :: from_raw ( unsafe {
387+ mlirDenseI32ArrayGet (
388+ builder. cx . raw ( ) ,
392389 3 ,
393390 [ observed. len ( ) as _ , time. is_some ( ) as _ , 0 ] . as_ptr ( ) ,
394391 )
395392 } ) ;
396- state. add_attribute ( "operand_segment_sizes" , vector_attr ) ;
393+ state. add_attribute ( "operand_segment_sizes" , dense_array_attr ) ;
397394 } )
398395 }
399396}
0 commit comments