Commit 10869f6c authored by yuguo960516yuguo's avatar yuguo960516yuguo
Browse files

hgemm fp32 hipfft.so

parent f0ef3442
......@@ -394,7 +394,7 @@ void* GetROCFFTDsoHandle() {
#if defined(__APPLE__) || defined(__OSX__)
return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "librocfft.dylib");
#else
return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "librocfft.so");
return GetDsoHandleFromSearchPath(FLAGS_rocm_dir, "libhipfft.so");
#endif
}
#endif
......
......@@ -738,7 +738,7 @@ inline void Blas<phi::GPUContext>::GEMM(CBLAS_TRANSPOSE transA,
C,
rocblas_datatype_f16_r,
N,
rocblas_datatype_f16_r);
rocblas_datatype_f32_r);
}
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