Skip to content

Commit 3fe0575

Browse files
authored
Merge pull request #13148 from dzhwinter/windows/math_compile
cuda math port
2 parents 1835e91 + 34757ef commit 3fe0575

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

paddle/fluid/operators/lod_tensor_to_array_op.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ limitations under the License. */
1616
#include "paddle/fluid/framework/op_registry.h"
1717
#include "paddle/fluid/operators/detail/safe_ref.h"
1818
#include "paddle/fluid/platform/device_context.h"
19+
#include "paddle/fluid/platform/port.h"
1920

2021
namespace paddle {
2122
namespace operators {

paddle/fluid/operators/math/CMakeLists.txt

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
1+
if (NOT WIN32)
12
add_subdirectory(detail)
3+
endif(NOT WIN32)
24

35
function(math_library TARGET)
46
# math_library is a function to create math library.
@@ -38,9 +40,13 @@ math_library(context_project DEPS im2col math_function)
3840
math_library(cross_entropy)
3941
math_library(cos_sim_functor)
4042
math_library(depthwise_conv)
41-
math_library(gru_compute DEPS activation_functions math_function)
4243
math_library(im2col)
44+
45+
if (NOT WIN32) # windows do not support avx functions yet.
46+
math_library(gru_compute DEPS activation_functions math_function)
4347
math_library(lstm_compute DEPS activation_functions)
48+
endif (NOT WIN32)
49+
4450
cc_library(blas SRCS blas.cc DEPS cblas framework_proto device_context)
4551
math_library(math_function DEPS blas)
4652
math_library(maxouting)
@@ -51,7 +57,9 @@ math_library(sequence_padding)
5157
math_library(sequence_pooling DEPS math_function)
5258
math_library(sequence_scale)
5359
math_library(softmax DEPS math_function)
60+
if (NOT WIN32)
5461
math_library(matrix_bit_code)
62+
endif (NOT WIN32)
5563
math_library(unpooling)
5664
math_library(vol2col)
5765

0 commit comments

Comments
 (0)