Tian Zheng issues

Repositories
Issues
Comments

Results 3 issues of


                                            Tian Zheng

[NVIDIA] TE Integration (Draft)

### PR types ### PR changes ### Description

contributor

stale

[NVIDIA] Need to support custom op mapping through python functions

**Please fill in the information below so that we can solve the problem quickly, Thanks !** **Describe the bug** A clear and concise description of what the bug is. After...

Fix test_flash_attention on sm_8x devices

### PR Category Operator Mechanism ### PR Types Bug fixes ### Description Fixes `test_flash_attention` on sm_8x devices. Reason: sm_86 & sm_89 does not support flash attention bwd kernel for D...

NVIDIA

contributor