Skip to content

[KB] Use --dtype instead of --bf16 - #176

Merged
rengolin merged 3 commits into
llvm:mainfrom
rengolin:kb_bench
Jun 2, 2026
Merged

[KB] Use --dtype instead of --bf16#176
rengolin merged 3 commits into
llvm:mainfrom
rengolin:kb_bench

Conversation

@rengolin

@rengolin rengolin commented Jun 2, 2026

Copy link
Copy Markdown
Member

As noted in the TODO, this PR makes use of #169 to allow conversion of models to an appropriate data type. For now, only F32 and BF16 are supported, but it should not be hard to make other data types available.

Also, fix the bf16 output return and add tests in CI.

As noted in the TODO, this PR makes use of llvm#169 to allow conversion of
models to an appropriate data type. For now, only F32 and BF16 are supported,
but it should not be hard to make other data types available.

Also, fix the bf16 output return and add tests in CI.
@rengolin
rengolin requested review from adam-smnk and tkarna June 2, 2026 14:45

@tkarna tkarna left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

Comment thread examples/KernelBench/test-kernel-bench.py
Comment thread tools/kernel-bench Outdated
rengolin added 2 commits June 2, 2026 16:10
I cannot reproduce the error on the same machine.
@rengolin
rengolin merged commit 2064122 into llvm:main Jun 2, 2026
3 checks passed
Comment thread tools/kernel-bench
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.

3 participants