From 5b08e953a2260f4a291b78de2e757bb4d54e1a52 Mon Sep 17 00:00:00 2001 From: PragmaTwice Date: Tue, 17 Dec 2024 11:16:46 +0800 Subject: [PATCH] [PJRT] Fix tensor element type for signed integers Signed-off-by: PragmaTwice --- integrations/pjrt/src/iree_pjrt/common/api_impl.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/integrations/pjrt/src/iree_pjrt/common/api_impl.cc b/integrations/pjrt/src/iree_pjrt/common/api_impl.cc index b0cf64ca67e0..cc8935619db2 100644 --- a/integrations/pjrt/src/iree_pjrt/common/api_impl.cc +++ b/integrations/pjrt/src/iree_pjrt/common/api_impl.cc @@ -103,19 +103,19 @@ iree_status_t MapElementTypeToMlirType(iree_hal_element_type_t element_type, *ty = "i1"; return iree_ok_status(); case IREE_HAL_ELEMENT_TYPE_SINT_4: - *ty = "si4"; + *ty = "i4"; return iree_ok_status(); case IREE_HAL_ELEMENT_TYPE_SINT_8: - *ty = "si8"; + *ty = "i8"; return iree_ok_status(); case IREE_HAL_ELEMENT_TYPE_SINT_16: - *ty = "si16"; + *ty = "i16"; return iree_ok_status(); case IREE_HAL_ELEMENT_TYPE_SINT_32: - *ty = "si32"; + *ty = "i32"; return iree_ok_status(); case IREE_HAL_ELEMENT_TYPE_SINT_64: - *ty = "si64"; + *ty = "i64"; return iree_ok_status(); case IREE_HAL_ELEMENT_TYPE_UINT_4: *ty = "ui4";