未验证 提交 325feca6 编写于 作者: Z Zhang Zheng 提交者: GitHub

Fix compile error in cublaslt (#51793)

上级 1816ceba
...@@ -17,9 +17,9 @@ limitations under the License. */ ...@@ -17,9 +17,9 @@ limitations under the License. */
#if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11060 #if defined(PADDLE_WITH_CUDA) && CUDA_VERSION >= 11060
#include <cuda_runtime_api.h> // NOLINT #include <cuda_runtime_api.h> // NOLINT
#include "cuda.h" // NOLINT #include "cuda.h" // NOLINT
#include "paddle/fluid/memory/malloc.h"
#include "paddle/phi/backends/dynload/cublasLt.h" #include "paddle/phi/backends/dynload/cublasLt.h"
#include "paddle/phi/common/amp_type_traits.h" #include "paddle/phi/common/amp_type_traits.h"
#include "paddle/phi/common/memory_utils.h"
#include "paddle/phi/kernels/autotune/gpu_timer.h" #include "paddle/phi/kernels/autotune/gpu_timer.h"
#include "paddle/phi/kernels/autotune/switch_autotune.h" #include "paddle/phi/kernels/autotune/switch_autotune.h"
#endif #endif
...@@ -389,7 +389,7 @@ struct MatmulWithCublasLt { ...@@ -389,7 +389,7 @@ struct MatmulWithCublasLt {
private: private:
static phi::Allocator::AllocationPtr GetWorkspace(const phi::GPUContext& ctx, static phi::Allocator::AllocationPtr GetWorkspace(const phi::GPUContext& ctx,
size_t workspace_size) { size_t workspace_size) {
return paddle::memory::Alloc( return phi::memory_utils::Alloc(
ctx.GetPlace(), ctx.GetPlace(),
workspace_size, workspace_size,
phi::Stream(reinterpret_cast<phi::StreamId>(ctx.stream()))); phi::Stream(reinterpret_cast<phi::StreamId>(ctx.stream())));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册