![](https://img-blog.csdnimg.cn/20190918140129601.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
机器学习
文章平均质量分 53
机器学习
MachineCYL
这个作者很懒,什么都没留下…
展开
-
梳理cuda算子编译与python调用的流程_以vllm为例
在这个例子中,编译后的输出文件是/data/caiyueliang/vllm/build/temp.linux-x86_64-cpython-310/csrc/cuda_utils_kernels.o。-DTORCH_API_INCLUDE_EXTENSION_H、-DTORCH_EXTENSION_NAME=_ext、-D_GLIBCXX_USE_CXX11_ABI=0: 这些是针对PyTorch扩展的特定宏定义。在上述代码中我们终于看到了vllm._C,该名字正是新定义的扩展的名字。原创 2024-02-02 18:54:39 · 1471 阅读 · 0 评论 -
【大数据】分布式机器学习平台
【大数据】分布式机器学习平台原创 2022-07-04 15:10:24 · 488 阅读 · 0 评论 -
【xgboost】贝叶斯自动调参代码
工作中,在用xgboost或者lightgbm模型进行模型训练的时候,经常用贝叶斯自动调参来搜索最优的参数。现在把相关的代码贴出来,供大家参考。原创 2022-07-06 16:07:00 · 2280 阅读 · 0 评论