mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-20 23:53:48 +08:00
a755a9e52c
Since currently `z1.h`, `z2.h` and `z3.h` are located under
`csrc/compile`, without this patch, torch hipify will fail to identify
these hipified headers on AMD platform:
```log
In file included from torch/include/ATen/cuda/CUDAEvent.h:3,
from deepspeed/ops/csrc/includes/deepcompile.h:16,
from deepspeed/ops/csrc/compile/z1.h:6,
from deepspeed/ops/csrc/compile/z1_hip.cpp:7:
torch/include/ATen/cuda/ATenCUDAGeneral.h:3:10: fatal error: cuda.h: No such file or directory
3 | #include <cuda.h>
| ^~~~~~~~
compilation terminated.
```
Signed-off-by: Hollow Man <hollowman@opensuse.org>