@@ -229,14 +229,14 @@ impl InstanceOp {
229
229
builder. build_with ( |builder, state| {
230
230
let num_inputs = inputs. into_iter ( ) . map ( |v| state. add_operand ( v) ) . count ( ) ;
231
231
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 ( ) ,
235
235
2 ,
236
236
[ num_inputs as _ , num_outputs as _ ] . as_ptr ( ) ,
237
237
)
238
238
} ) ;
239
- state. add_attribute ( "operand_segment_sizes" , vector_attr ) ;
239
+ state. add_attribute ( "operand_segment_sizes" , dense_array_attr ) ;
240
240
state. add_attribute ( "name" , get_string_attr ( builder. cx , name) ) ;
241
241
state. add_attribute ( "callee" , get_flat_symbol_ref_attr ( builder. cx , module) ) ;
242
242
} )
@@ -383,17 +383,14 @@ impl WaitOp {
383
383
if let Some ( time) = time {
384
384
state. add_operand ( time) ;
385
385
}
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 ( ) ,
392
389
3 ,
393
390
[ observed. len ( ) as _ , time. is_some ( ) as _ , 0 ] . as_ptr ( ) ,
394
391
)
395
392
} ) ;
396
- state. add_attribute ( "operand_segment_sizes" , vector_attr ) ;
393
+ state. add_attribute ( "operand_segment_sizes" , dense_array_attr ) ;
397
394
} )
398
395
}
399
396
}
0 commit comments