Skip to content

Commit d5791b0

Browse files
d0kcopybara-github
authored andcommitted
[XLA] Migrate off deprecated non-opaque LLVM pointer types
These are being removed in upstream MLIR, blocking imports of new MLIR versions. PiperOrigin-RevId: 580006151
1 parent adee868 commit d5791b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

xla/service/gpu/ir_emitter_triton.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -776,7 +776,7 @@ void StripParameterAddressSpaces(mlir::RewriterBase& rewriter,
776776
auto ptr_ty = type.dyn_cast<ml::LLVMPointerType>();
777777
if (!ptr_ty) return type;
778778
if (ptr_ty.getAddressSpace() != mn::kGlobalMemorySpace) return type;
779-
return ml::LLVMPointerType::get(ptr_ty.getElementType());
779+
return ml::LLVMPointerType::get(ptr_ty.getContext());
780780
}));
781781
ml::LLVMFunctionType generic_func_ty =
782782
func_ty.clone(generic_func_params, func_ty.getReturnTypes());

0 commit comments

Comments
 (0)