InfiniTensor/include/kunlun
zhangyue 985d0dee5f
Kunlun dist op (#225)
* kunlun dist inference fix

* kunlun distributed

* 添加昆仑芯分布式脚本以及解决运行llama遇到的问题

* set -j8

* format

* move run_pytorch.py int o cuda/

* update notes

---------

Co-authored-by: weijie01 <weijie01@baidu.com>
Co-authored-by: wanghailu <wanghailu0717@163.com>
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2024-04-23 15:46:25 +08:00
..
kunlun_act_type.h XCCL support (#171) 2024-02-29 11:48:35 +08:00
kunlun_common.h XCCL support (#171) 2024-02-29 11:48:35 +08:00
kunlun_kernel_without_config.h Xpu (#82) 2023-10-16 10:57:08 +08:00
kunlun_runtime.h Kunlun dist op (#225) 2024-04-23 15:46:25 +08:00
operator_timer.h Xpu (#82) 2023-10-16 10:57:08 +08:00
xccl_communicator.h Kunlun dist op (#225) 2024-04-23 15:46:25 +08:00