DTK开发
Deepspeed 报错
2024-08-26 12:39:08
134

/usr/local/lib/python3.10/site-packages/deepspeed/ops/adam/fused_adam.py:96: UserWarning: The torch.cuda.DtypeTensor constructors are no longer recommended. It's best to use methods such as torch.tensor(data, dtype=, device='cuda') to create tensors. (Triggered internally at /data/jenkins_workspace/workspace/pytorch@4/torch/csrc/tensor/python_tensor.cpp:83.)
self._dummy_overflow_buf = get_accelerator().IntTensor([0])
Launch params (512, 1, 1) are larger than launch bounds (256) for kernel Z25multi_tensor_apply_kernelI18TensorListMetadataILi4EE11AdamFunctorIfEJffffff10adamMode_tfEEviPViT_T0_DpT1 please add launch_bounds to kernel define or use --gpu-max-threads-per-block recompile program !
Launch params (512, 1, 1) are larger than launch bounds (256) for kernel Z25multi_tensor_apply_kernelI18TensorListMetadataILi4EE11AdamFunctorIfEJffffff10adamMode_tfEEviPViT_T0_DpT1 please add launch_bounds to kernel define or use --gpu-max-threads-per-block recompile program !
Launch params (512, 1, 1) are larger than launch bounds (256) for kernel Z25multi_tensor_apply_kernelI18TensorListMetadataILi4EE11AdamFunctorIfEJffffff10adamMode_tfEEviPViT_T0_DpT1 please add launch_bounds to kernel define or use --gpu-max-threads-per-block recompile program !

attachmentId-image.png

全部评论(2)
只看楼主
网
网友ycHhjK
1楼
这里没有报错,只是warning。第一个是deepspeed调用torch接口比较老,这是官方自己的调用。第二种warning,后续版本可以加入编译参数去掉。
2024/8/27
东
东风
回复 @网友ycHhjK

我在ollama中遇到了这个问题,现在的AMD卡的llvm已经将launch_bounds改成了1024,不知道为什么DCU还停留在古早的256,连K100也是这样。

图片中是github网友对此的评价
2025/2/19
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据