Commit b5499578 authored by yuguo960516yuguo's avatar yuguo960516yuguo
Browse files

hgemm fp32 hipfft.so

parent ca73620b
...@@ -394,7 +394,7 @@ void* GetROCFFTDsoHandle() { ...@@ -394,7 +394,7 @@ void* GetROCFFTDsoHandle() {
#if defined(__APPLE__) || defined(__OSX__) #if defined(__APPLE__) || defined(__OSX__)
return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "librocfft.dylib"); return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "librocfft.dylib");
#else #else
return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "librocfft.so"); return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "libhipfft.so");
#endif #endif
} }
#endif #endif
......
...@@ -738,7 +738,7 @@ inline void Blas<phi::GPUContext>::GEMM(CBLAS_TRANSPOSE transA, ...@@ -738,7 +738,7 @@ inline void Blas<phi::GPUContext>::GEMM(CBLAS_TRANSPOSE transA,
C, C,
rocblas_datatype_f16_r, rocblas_datatype_f16_r,
N, N,
rocblas_datatype_f16_r); rocblas_datatype_f32_r);
} }
template <> template <>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment