Skip to content

benchmark-v2: minor fix for benchmark-v2, kernel is not correctly called#43428

Merged
remi-or merged 8 commits intohuggingface:mainfrom
sywangyi:benchmark-v2
Jan 31, 2026
Merged

benchmark-v2: minor fix for benchmark-v2, kernel is not correctly called#43428
remi-or merged 8 commits intohuggingface:mainfrom
sywangyi:benchmark-v2

Conversation

@sywangyi
Copy link
Contributor

kernels is not correctly called in benchmark_v2, should register before kernelize, see

register_kernel_mapping_transformers()
, use 'use_kernels' in AutoModelForCausalLM.from_pretrained could handle register kernels and kernelize

@sywangyi
Copy link
Contributor Author

@remi-or , pls help review, thx very much.

Signed-off-by: Wang, Yi <yi.a.wang@intel.com>
Signed-off-by: Wang, Yi <yi.a.wang@intel.com>
@github-actions
Copy link
Contributor

View the CircleCI Test Summary for this PR:

https://huggingface.co/spaces/transformers-community/circle-ci-viz?pr=43428&sha=0c40fd

Copy link
Collaborator

@remi-or remi-or left a comment

Choose a reason for hiding this comment

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

Thanks for the contribution! A few nits to address but otherwise we should be good.

Signed-off-by: Wang, Yi <yi.a.wang@intel.com>
Copy link
Collaborator

@remi-or remi-or left a comment

Choose a reason for hiding this comment

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

LGTM, thanks for the contribution and the insights!

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@yao-matrix
Copy link
Contributor

@SunMarc , could you help merge it? Thx very much.

@remi-or remi-or merged commit 1722163 into huggingface:main Jan 31, 2026
15 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.

4 participants