search

ROCm / hipify_torch

MIT License
18 stars 10 forks source link

hipify DeepSpeed error #35

Open fwang2 opened 2 years ago

fwang2 commented 2 years ago

hi there -

I am trying to build proper extension for DeepSpeed, for example, DS_CPU_ADAM=1, and it bailed out at the following error:

**cc1plus: warning: command-line option '-Wstrict-prototypes' is valid for C/ObjC but not for C++
In file included from csrc/adam/cpu_adam_hip.cpp:2:
csrc/includes/cpu_adam.h:6:10: fatal error: cuda_fp16.h: No such file or directory
    6 | #include <cuda_fp16.h>
      |          ^~~~~~~~~~~~~
compilation terminated.

Noted that cpu_adam_hip.cpp included cpu_adam.h, not the cpu_adam_hip.h it generated during hipify process. Is this the correct behavior or a bug? Both DeepSpeed and torch_hipify are using the head of the repo.

// !!! This is a file automatically generated by hipify!!!
#include "cpu_adam.h"
#include <hip/hip_runtime_api.h>