[Mlir-commits] [mlir] [mlir][amdgpu] Add tensor load store operations (PR #170918)
Erick Ochoa Lopez
llvmlistbot at llvm.org
Wed Dec 17 06:56:06 PST 2025
================
@@ -3306,6 +3327,24 @@ void mlir::populateAMDGPUTypeAndAttributeConversions(
Type i32 = IntegerType::get(type.getContext(), 32);
return typeConverter.convertType(VectorType::get(4, i32));
});
+ typeConverter.addConversion(
+ [&](TDMDescriptorType type,
+ SmallVectorImpl<Type> &result) -> std::optional<LogicalResult> {
+ Type i32 = IntegerType::get(type.getContext(), 32);
+ Type v4i32 = typeConverter.convertType(VectorType::get(4, i32));
+ Type v8i32 = typeConverter.convertType(VectorType::get(8, i32));
+ llvm::append_values(result, v4i32, v8i32, v4i32, v4i32);
+ return success();
+ });
+
+ auto addUnrealizedCast = [](OpBuilder &builder, TypeRange types,
+ ValueRange inputs,
+ Location loc) -> SmallVector<Value> {
+ auto cast = UnrealizedConversionCastOp::create(builder, loc, types, inputs);
+ return cast.getResults();
+ };
+
+ typeConverter.addTargetMaterialization(addUnrealizedCast);
----------------
amd-eochoalo wrote:
Ah thank you, I didn't notice. Let's revert it now and I'll fix it later. One second.
https://github.com/llvm/llvm-project/pull/170918
More information about the Mlir-commits
mailing list