Skip to content

Commit 059155c

Browse files
Regenerate MLIR Bindings (#2799)
Co-authored-by: enzyme-ci-bot[bot] <78882869+enzyme-ci-bot[bot]@users.noreply.github.com>
1 parent 9a6b0f0 commit 059155c

2 files changed

Lines changed: 106 additions & 5 deletions

File tree

src/mlir/Dialects/Affine.jl

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -543,17 +543,16 @@ In the above example, `%linear_index` conceptually holds the following:
543543
function linearize_index(
544544
multi_index::Vector{Value},
545545
dynamic_basis::Vector{Value};
546-
linear_index=nothing::Union{Nothing,IR.Type},
546+
linear_index::IR.Type,
547547
static_basis,
548548
location=Location(),
549549
)
550-
op_ty_results = IR.Type[]
550+
op_ty_results = IR.Type[linear_index,]
551551
operands = Value[multi_index..., dynamic_basis...]
552552
owned_regions = Region[]
553553
successors = Block[]
554554
attributes = NamedAttribute[NamedAttribute("static_basis", static_basis),]
555555
push!(attributes, operandsegmentsizes([length(multi_index), length(dynamic_basis)]))
556-
!isnothing(linear_index) && push!(op_ty_results, linear_index)
557556

558557
return create_operation(
559558
"affine.linearize_index",
@@ -562,8 +561,8 @@ function linearize_index(
562561
owned_regions,
563562
successors,
564563
attributes,
565-
results=(length(op_ty_results) == 0 ? nothing : op_ty_results),
566-
result_inference=(length(op_ty_results) == 0 ? true : false),
564+
results=op_ty_results,
565+
result_inference=false,
567566
)
568567
end
569568

src/mlir/libMLIR_h.jl

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13562,6 +13562,108 @@ function mlirMosaicGpuSwizzleTransformAttrGetTypeID()
1356213562
@ccall mlir_c.mlirMosaicGpuSwizzleTransformAttrGetTypeID()::MlirTypeID
1356313563
end
1356413564

13565+
function mlirMosaicGpuIsAWGSplatFragLayoutAttr(attr)
13566+
@ccall mlir_c.mlirMosaicGpuIsAWGSplatFragLayoutAttr(attr::MlirAttribute)::Bool
13567+
end
13568+
13569+
function mlirMosaicGpuWGSplatFragLayoutAttrGetTypeID()
13570+
@ccall mlir_c.mlirMosaicGpuWGSplatFragLayoutAttrGetTypeID()::MlirTypeID
13571+
end
13572+
13573+
function mlirMosaicGpuWGSplatFragLayoutAttrGet(ctx, shape)
13574+
@ccall mlir_c.mlirMosaicGpuWGSplatFragLayoutAttrGet(
13575+
ctx::MlirContext, shape::MlirAttribute
13576+
)::MlirAttribute
13577+
end
13578+
13579+
function mlirMosaicGpuWGSplatFragLayoutAttrGetShape(attr)
13580+
@ccall mlir_c.mlirMosaicGpuWGSplatFragLayoutAttrGetShape(
13581+
attr::MlirAttribute
13582+
)::MlirAttribute
13583+
end
13584+
13585+
function mlirMosaicGpuIsAWGStridedFragLayoutAttr(attr)
13586+
@ccall mlir_c.mlirMosaicGpuIsAWGStridedFragLayoutAttr(attr::MlirAttribute)::Bool
13587+
end
13588+
13589+
function mlirMosaicGpuWGStridedFragLayoutAttrGetTypeID()
13590+
@ccall mlir_c.mlirMosaicGpuWGStridedFragLayoutAttrGetTypeID()::MlirTypeID
13591+
end
13592+
13593+
function mlirMosaicGpuWGStridedFragLayoutAttrGet(ctx, shape, vector_size)
13594+
@ccall mlir_c.mlirMosaicGpuWGStridedFragLayoutAttrGet(
13595+
ctx::MlirContext, shape::MlirAttribute, vector_size::Int32
13596+
)::MlirAttribute
13597+
end
13598+
13599+
function mlirMosaicGpuWGStridedFragLayoutAttrGetShape(attr)
13600+
@ccall mlir_c.mlirMosaicGpuWGStridedFragLayoutAttrGetShape(
13601+
attr::MlirAttribute
13602+
)::MlirAttribute
13603+
end
13604+
13605+
function mlirMosaicGpuWGStridedFragLayoutAttrGetVectorSize(attr)
13606+
@ccall mlir_c.mlirMosaicGpuWGStridedFragLayoutAttrGetVectorSize(
13607+
attr::MlirAttribute
13608+
)::Int32
13609+
end
13610+
13611+
function mlirMosaicGpuIsAReplicatedAttr(attr)
13612+
@ccall mlir_c.mlirMosaicGpuIsAReplicatedAttr(attr::MlirAttribute)::Bool
13613+
end
13614+
13615+
function mlirMosaicGpuReplicatedAttrGetTypeID()
13616+
@ccall mlir_c.mlirMosaicGpuReplicatedAttrGetTypeID()::MlirTypeID
13617+
end
13618+
13619+
function mlirMosaicGpuReplicatedAttrGet(ctx, times)
13620+
@ccall mlir_c.mlirMosaicGpuReplicatedAttrGet(
13621+
ctx::MlirContext, times::Int32
13622+
)::MlirAttribute
13623+
end
13624+
13625+
function mlirMosaicGpuReplicatedAttrGetTimes(attr)
13626+
@ccall mlir_c.mlirMosaicGpuReplicatedAttrGetTimes(attr::MlirAttribute)::Int32
13627+
end
13628+
13629+
function mlirMosaicGpuIsATiledLayoutAttr(attr)
13630+
@ccall mlir_c.mlirMosaicGpuIsATiledLayoutAttr(attr::MlirAttribute)::Bool
13631+
end
13632+
13633+
function mlirMosaicGpuTiledLayoutAttrGetTypeID()
13634+
@ccall mlir_c.mlirMosaicGpuTiledLayoutAttrGetTypeID()::MlirTypeID
13635+
end
13636+
13637+
function mlirMosaicGpuTiledLayoutAttrGet(ctx, tiling, warp_dims, lane_dims, vector_dim)
13638+
@ccall mlir_c.mlirMosaicGpuTiledLayoutAttrGet(
13639+
ctx::MlirContext,
13640+
tiling::MlirAttribute,
13641+
warp_dims::MlirAttribute,
13642+
lane_dims::MlirAttribute,
13643+
vector_dim::Int32,
13644+
)::MlirAttribute
13645+
end
13646+
13647+
function mlirMosaicGpuTiledLayoutAttrGetTiling(attr)
13648+
@ccall mlir_c.mlirMosaicGpuTiledLayoutAttrGetTiling(attr::MlirAttribute)::MlirAttribute
13649+
end
13650+
13651+
function mlirMosaicGpuTiledLayoutAttrGetWarpDims(attr)
13652+
@ccall mlir_c.mlirMosaicGpuTiledLayoutAttrGetWarpDims(
13653+
attr::MlirAttribute
13654+
)::MlirAttribute
13655+
end
13656+
13657+
function mlirMosaicGpuTiledLayoutAttrGetLaneDims(attr)
13658+
@ccall mlir_c.mlirMosaicGpuTiledLayoutAttrGetLaneDims(
13659+
attr::MlirAttribute
13660+
)::MlirAttribute
13661+
end
13662+
13663+
function mlirMosaicGpuTiledLayoutAttrGetVectorDim(attr)
13664+
@ccall mlir_c.mlirMosaicGpuTiledLayoutAttrGetVectorDim(attr::MlirAttribute)::Int32
13665+
end
13666+
1356513667
function mlirMosaicGpuIsACopyPartitionAttr(attr)
1356613668
@ccall mlir_c.mlirMosaicGpuIsACopyPartitionAttr(attr::MlirAttribute)::Bool
1356713669
end

0 commit comments

Comments
 (0)