changes for tensorflow

bragadeesh · Mar 23, 2018 · 38e445b · 38e445b
1 parent dfcad74
commit 38e445b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/library/include/rocfft/hipfft.h b/library/include/rocfft/hipfft.h
@@ -6,6 +6,7 @@
 #define __HIPFFT_H__
 
 #include <hip/hip_runtime_api.h>
+#include <hip/hip_complex.h>
 
 #define DLL_PUBLIC __attribute__ ((visibility ("default")))
 
@@ -46,8 +47,8 @@ typedef enum hipfftType_t {
 
 typedef struct hipfftHandle_t *hipfftHandle;
 
-typedef float hipfftComplex[2];
-typedef double hipfftDoubleComplex[2];
+typedef hipComplex hipfftComplex;
+typedef hipDoubleComplex hipfftDoubleComplex;
 typedef float  hipfftReal;
 typedef double hipfftDoubleReal;