Skip to content

Conversation

@gongchensu
Copy link
Collaborator

@gongchensu gongchensu commented Oct 30, 2025

  1. 实现gemm的python层,增加gemm测试
  2. matmul_infiniop.cc改成了gemm_infiniop.cc,matmul的调用改为调用gemm只不过alpha默认为1,beta默认为0,避免重复代码
    Nvidia测试:
image CPU测试: image

matmul修改后测试:
image
image

@gongchensu gongchensu force-pushed the feature/add_gemm_python_api branch from 5722561 to f8b639e Compare October 30, 2025 07:40
@gongchensu gongchensu self-assigned this Oct 30, 2025
@gongchensu gongchensu linked an issue Oct 30, 2025 that may be closed by this pull request
@gongchensu gongchensu requested a review from wooway777 October 30, 2025 09:28
@pengcheng888 pengcheng888 self-requested a review October 31, 2025 02:13
@gongchensu gongchensu force-pushed the feature/add_gemm_python_api branch from f5d1251 to 001141c Compare October 31, 2025 04:32
@pengcheng888 pengcheng888 merged commit 943b38e into InfiniTensor:main Oct 31, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[DEV] 为gemm增加python接口

2 participants