Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cambricon: update mlu with master #46

Closed
wants to merge 19 commits into from
Closed

cambricon: update mlu with master #46

wants to merge 19 commits into from

Conversation

FuncSherl
Copy link
Collaborator

update mlu with master

iclementine and others added 19 commits May 27, 2024 09:55
* refactor codegen for poinwise
1. support 0d task space;
2. support specifying output dtypes;
3. support non-tensor arguments to the scalar function;
4. add more type annotation in the generated code.
* Add fused apply_rotary_pos_emb

* Add contiguous check for cos/sin and refine shape check

* refine by comment
* [fix] Temporarily use upcasting to make prod support bf16

* add bfloat16 type to op_accu_test

* simplify test cases for all, any
…unction to ensure that they are never wrapped into tl.constexpr. (#37)
* [test] skip testing all_dim/all_dims/any_dim/any_dims for pytorch < 2.2.0

* [operator] implement scalar func of bitwise_and/bitwise_or/clamp

* [operator] binary pointwise operators by codegen. eq/ne/ge/gt/le/lt/isinf/isnan.

* [operator] matrix-vector multiplication
* add cross_entropy_loss

* clean code

* modify default value of reduction

---------

Co-authored-by: jiangbin <[email protected]>
1. reorganized test files in examples, benchmark, tests respectively
2. divide operators into pointwise, reduction, bias and so on
3. implement v1 bitwise ops add, sub, mul, div and pow by codegen
[ci/cd] Add python-test config yaml for Unit Test
* modify name && add reduce function

* add reduce none

* add test

* clean code

* add reduce enum

* Replacing the enum interface with Intenum & add illegal detection of reduction

---------

Co-authored-by: Jiang Bin <[email protected]>
[CI/CD] split unit test and model test
@FuncSherl FuncSherl closed this Jun 3, 2024
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.

7 participants