Home Explore Help
Register Sign In
david
/
aphrodite-engine
mirror of https://github.com/PygmalionAI/aphrodite-engine
1
0
Fork 0
Files Issues 0 Wiki
Tree: c154578c97
Branches Tags
aphrodite-engin...
/
rocm_patch
/
rocm_bf16.patch

rocm_bf16.patch 545 B
History Raw

123456789101112131415 
  1. --- amd_hip_bf16.h	2024-02-06 18:28:58.268699142 +0000
    2. 

  2. +++ amd_hip_bf16.h.new	2024-02-06 18:28:31.988647133 +0000
    3. 

  3. @@ -90,10 +90,10 @@
    4. 

  4.  #include "math_fwd.h"              // ocml device functions
    5. 

  5.  
    6. 

  6.  #if defined(__HIPCC_RTC__)
    7. 

  7. -#define __HOST_DEVICE__ __device__
    8. 

  8. +#define __HOST_DEVICE__ __device__ static
    9. 

  9.  #else
    10. 

  10.  #include <climits>
    11. 

  11. -#define __HOST_DEVICE__ __host__ __device__
    12. 

  12. +#define __HOST_DEVICE__ __host__ __device__ static inline
    13. 

  13.  #endif
    14. 

  14.  
    15. 

  15.  // Since we are using unsigned short to represent data in bfloat16, it can be of different sizes on
    16.