- 14 Jul, 2025 1 commit
-
-
YdrMaster authored
Signed-off-by:YdrMaster <ydrml@hotmail.com>
-
- 11 Jul, 2025 1 commit
-
-
YdrMaster authored
Signed-off-by:YdrMaster <ydrml@hotmail.com>
-
- 10 Jul, 2025 1 commit
-
-
YdrMaster authored
Signed-off-by:YdrMaster <ydrml@hotmail.com>
-
- 01 Jul, 2025 1 commit
-
-
蒋帅宏(Shuaihong_Jiang) authored
* issue/254: 添加算子在CPU和CUDA上对BF16的支持,并增加相应的测试代码 * issue/254: 将修改后的算子格式化后重新提交 * 修改与最新main的冲突 * 解决冲突后rms_norm原本的精度过不了了,现在由 {"atol": 5e-3, "rtol": 5e-3}更改为 {"atol": 8e-3, "rtol": 8e-3} * rms_norm在debug模式下FP16的测试用例失败了(本地测试能通过,github上过不了), 所以将容差增大了两倍进行测试 * 将rms_normd的测试输入缩放0.5,将容差改回原始值来进行ci测试 * issue/254: 1.使用CHECK_DTYPE宏来进行数据类型检验 2.在test的utils.py中添加了设备对BF16支持的检验 * issue/254: rms_norm测试fp16容差由 torch.float16: {"atol": 1e-3, "rtol": 1e-3}, 改为torch.float16: {"atol": 2e-3, "rtol": 2e-3}, 并删除对输入0.5的放缩 * issue/254: 在utils.py中debug方法和debug_all方法中 添加了对BF16的特判 * 修改支持BF16测试的设备类型检查方法 * 修改支持BF16测试的设备检查 * issue/254: reduce redundancy in rms_norm.py * issue/254: add back the missing comment in rms_norm.py * issue/254: add fp32 tolerance condition in causal_softmax.py --------- Co-authored-by:Zimin Li <coollizimin@gmail.com>
-
- 04 Jun, 2025 3 commits
- 30 Apr, 2025 4 commits
- 29 Apr, 2025 1 commit
-
-
zhangyue authored
-
- 25 Apr, 2025 2 commits
-
-
Graylatzhou authored
-
Graylatzhou authored
-
- 23 Apr, 2025 1 commit
-
-
Zimin Li authored
issue/194: fix meta mem size allocation and change cpu DeviceImpl::create to return INFINI_STATUS_NOT_IMPLEMENTED
-
- 22 Apr, 2025 1 commit
-
-
PanZezhong authored
-
- 15 Apr, 2025 3 commits
-
-
Zimin Li authored
issue/127: change meta within ElementwiseInfo to std::vector<size_t> for correct alignment and change the reference name of the Opaque struct to Opaque instead of struct Opaque
-
Zimin Li authored
-
Zimin Li authored
issue/127: Add arguments to CREATE_ELEMENTWISE_PLATFORM_DESCRIPTOR macros for indirecting variable names, change DeviceImpl to use Result for the return type of the create function, change CEIL_DIV
-
- 14 Apr, 2025 10 commits
-
-
Zimin Li authored
-
Zimin Li authored
-
Zimin Li authored
issue/127: Optimize elementwise CUDA code by removing redundancy, change/correct kernel logic when all inputs have the same dtype
-
Zimin Li authored
issue/127: Refactor ElementwiseInfo, refactor elementwise to use workspace for storing meta, fix misc. issues
-
Zimin Li authored
issue/127: fix CUDA mix-precision broadcasting input mismatch issue, adjust comment structure and template variable order
-
Zimin Li authored
enable_if, remove std::move() in elementwise_cpu.h, add <array> inclusion
-
Zimin Li authored
issue/127: refactor ElementwiseInfo to use utils::Result, change elementwise calcualte and calculateImpl to return infiniStatus_t, add CHECK_CUDA to cuda function calls
-
Zimin Li authored
issue/127: modify swiglu test to correctly handle broadcast scenarios, add two broadcast testcases, correct elementwise cpu mix-precision implementation
-
Zimin Li authored
issue/127: refactor elementwise framework, complete CUDA implementation, refactor swiglu using the generic elementwise framework
-
Zimin Li authored
-