Skip to content

feat: add the implementation of Gemm operator on Cambricon#7

Open
bitzyz wants to merge 1 commit intofeat/dev-infrafrom
feat/dev-gemm-cambricon
Open

feat: add the implementation of Gemm operator on Cambricon#7
bitzyz wants to merge 1 commit intofeat/dev-infrafrom
feat/dev-gemm-cambricon

Conversation

@bitzyz
Copy link

@bitzyz bitzyz commented Mar 2, 2026

  1. Add Cambricon CMake options
  2. Implement Cambricon GEMM operator in src/cambricon/gemm/cnblas.h
  3. Add MLU skip logic in tests/test_gemm.py (transpose operations not currently supported)
  4. Add Cambricon test in examples/gemm/gemm.cc
  5. Add protected member variable workspace_size_ to Operator class
  6. Add Cmabricon common utils_func in src/cambricon/common.h

@bitzyz bitzyz self-assigned this Mar 2, 2026
@bitzyz bitzyz force-pushed the feat/dev-gemm-cambricon branch from 07207ac to a39d532 Compare March 2, 2026 07:09
@bitzyz
Copy link
Author

bitzyz commented Mar 2, 2026

image

@bitzyz bitzyz requested a review from voltjia March 2, 2026 07:10
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.

1 participant