Skip to content

[Runtime][Python] Convert numpy.uint8 to ABI type i8#24524

Merged
AGindinson merged 1 commit into
iree-org:mainfrom
Manewing:walbroel.iree_benchmark_runtime_numpy_uint8_fix
May 26, 2026
Merged

[Runtime][Python] Convert numpy.uint8 to ABI type i8#24524
AGindinson merged 1 commit into
iree-org:mainfrom
Manewing:walbroel.iree_benchmark_runtime_numpy_uint8_fix

Conversation

@Manewing
Copy link
Copy Markdown
Contributor

@Manewing Manewing commented May 21, 2026

Benchmarks are currently failing because of "KeyError: dtype(uint8)" these are fixed by simply converting numpy.uint8 to i8.

Also added other missing data types for numpy, i.e. uint16 and uint32 and added a test case to ensure the benchmark argument creation works for all cases

@Manewing Manewing requested a review from stellaraccident as a code owner May 21, 2026 14:53
Copy link
Copy Markdown
Contributor

@AGindinson AGindinson left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Does it make sense to add uint16/uint32, just in case?

@maxbartel maxbartel changed the title iree/runtime/benchmark: fix: convert numpy.uint8 to ABI type i8 [Runtime][Python] iree/runtime/benchmark: fix: convert numpy.uint8 to ABI type i8 May 26, 2026
Signed-off-by: Florian Walbroel <walbroel@roofline.ai>
@Manewing Manewing force-pushed the walbroel.iree_benchmark_runtime_numpy_uint8_fix branch from 1918f94 to 0bb7e7b Compare May 26, 2026 13:44
@Manewing Manewing requested a review from AGindinson May 26, 2026 13:45
@AGindinson AGindinson changed the title [Runtime][Python] iree/runtime/benchmark: fix: convert numpy.uint8 to ABI type i8 [Runtime][Python] Convert numpy.uint8 to ABI type i8 May 26, 2026
Copy link
Copy Markdown
Contributor

@AGindinson AGindinson left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, appreciate the updates!

@AGindinson AGindinson enabled auto-merge (squash) May 26, 2026 14:15
@AGindinson AGindinson merged commit 48e60e5 into iree-org:main May 26, 2026
61 of 62 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants