FIX: header file

cblas_new
liaogang 8 years ago
parent 2680dca9c8
commit 90c4cd8323

@ -18,8 +18,8 @@ limitations under the License. */
#include <cstring>
#include <memory>
#include "paddle/framework/ddim.h"
#include "paddle/platform/enforce.h"
#include "paddle/memory/memory.h"
#include "paddle/platform/enforce.h"
#include "paddle/platform/place.h"
namespace paddle {

@ -12,15 +12,16 @@ limitations under the License. */
#pragma once
#include "paddle/platform/enforce.h"
#include "paddle/platform/place.h"
#ifndef PADDLE_ONLY_CPU
#include "paddle/platform/dynload/cublas.h"
#include "paddle/platform/dynload/cudnn.h"
#include "paddle/platform/dynload/curand.h"
#include "paddle/platform/error.h"
#include "paddle/platform/gpu_info.h"
#define EIGEN_USE_GPU
#endif
#include <paddle/platform/place.h>
#include <memory>
#include <unsupported/Eigen/CXX11/Tensor>

Loading…
Cancel
Save