Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit ab93b2d

Browse files
hanke580hanke580
committed
* Fix windows oversize by adding files
1 parent 7463484 commit ab93b2d

File tree

5 files changed

+221
-140
lines changed

5 files changed

+221
-140
lines changed

src/api/operator/numpy/np_elemwise_broadcast_op_extended.cc renamed to src/api/operator/numpy/np_elemwise_broadcast_op_extended_sec.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818
*/
1919

2020
/*!
21-
* \file np_elemwise_broadcast_op.cc
22-
* \brief Implementation of the API of functions in src/operator/numpy/np_elemwise_broadcast_op.cc
21+
* \file np_elemwise_broadcast_op_extended_sec.cc
22+
* \brief Implementation of the API of functions in src/operator/numpy/np_elemwise_broadcast_op_extended_sec.cc
2323
*/
2424
#include <mxnet/api_registry.h>
2525
#include <mxnet/runtime/packed_func.h>

‎src/operator/numpy/np_elemwise_broadcast_op.cc‎

Lines changed: 0 additions & 93 deletions
Original file line numberDiff line numberDiff line change
@@ -218,98 +218,5 @@ MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_rpower_scalar)
218218
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::rpower>)
219219
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseOut{"_backward_rpower_scalar"});
220220

221-
MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(_npi_fmax)
222-
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::fmax>)
223-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmax"});
224-
225-
NNVM_REGISTER_OP(_backward_npi_fmax)
226-
.set_num_inputs(3)
227-
.set_num_outputs(2)
228-
.set_attr<nnvm::TIsBackward>("TIsBackward", true)
229-
.set_attr<nnvm::FInplaceOption>("FInplaceOption",
230-
[](const NodeAttrs& attrs){
231-
return std::vector<std::pair<int, int> >{{0, 1}};
232-
})
233-
.set_attr<FResourceRequest>("FResourceRequest",
234-
[](const NodeAttrs& attrs) {
235-
return std::vector<ResourceRequest>{ResourceRequest::kTempSpace};
236-
})
237-
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastBackwardUseIn<cpu, mshadow_op::ge,
238-
mshadow_op::lt>);
239-
240-
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_fmax_scalar)
241-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::fmax>)
242-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmax_scalar"});
243-
244-
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_fmax_scalar)
245-
.add_argument("scalar", "float", "scalar value")
246-
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
247-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::ge>);
248-
249-
MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(_npi_fmin)
250-
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::fmin>)
251-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmin"});
252-
253-
NNVM_REGISTER_OP(_backward_npi_fmin)
254-
.set_num_inputs(3)
255-
.set_num_outputs(2)
256-
.set_attr<nnvm::TIsBackward>("TIsBackward", true)
257-
.set_attr<nnvm::FInplaceOption>("FInplaceOption",
258-
[](const NodeAttrs& attrs){
259-
return std::vector<std::pair<int, int> >{{0, 1}};
260-
})
261-
.set_attr<FResourceRequest>("FResourceRequest",
262-
[](const NodeAttrs& attrs) {
263-
return std::vector<ResourceRequest>{ResourceRequest::kTempSpace};
264-
})
265-
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastBackwardUseIn<cpu, mshadow_op::le,
266-
mshadow_op::gt>);
267-
268-
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_fmin_scalar)
269-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::fmin>)
270-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmin_scalar"});
271-
272-
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_fmin_scalar)
273-
.add_argument("scalar", "float", "scalar value")
274-
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
275-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::le>);
276-
277-
MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(_npi_fmod)
278-
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::fmod>)
279-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmod"});
280-
281-
NNVM_REGISTER_OP(_backward_npi_fmod)
282-
.set_num_inputs(3)
283-
.set_num_outputs(2)
284-
.set_attr<nnvm::TIsBackward>("TIsBackward", true)
285-
.set_attr<nnvm::FInplaceOption>("FInplaceOption",
286-
[](const NodeAttrs& attrs){
287-
return std::vector<std::pair<int, int> >{{0, 1}};
288-
})
289-
.set_attr<FResourceRequest>("FResourceRequest",
290-
[](const NodeAttrs& attrs) {
291-
return std::vector<ResourceRequest>{ResourceRequest::kTempSpace};
292-
})
293-
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastBackwardUseIn<cpu, mshadow_op::mod_grad,
294-
mshadow_op::mod_rgrad>);
295-
296-
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_fmod_scalar)
297-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::fmod>)
298-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmod_scalar"});
299-
300-
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_fmod_scalar)
301-
.add_argument("scalar", "float", "scalar value")
302-
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
303-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::mod_grad>);
304-
305-
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_rfmod_scalar)
306-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::rfmod>)
307-
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_rfmod_scalar"});
308-
309-
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_rfmod_scalar)
310-
.add_argument("scalar", "float", "scalar value")
311-
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
312-
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::rmod_grad>);
313-
314221
} // namespace op
315222
} // namespace mxnet

‎src/operator/numpy/np_elemwise_broadcast_op.cu‎

Lines changed: 0 additions & 45 deletions
Original file line numberDiff line numberDiff line change
@@ -111,50 +111,5 @@ NNVM_REGISTER_OP(_npi_power_scalar)
111111
NNVM_REGISTER_OP(_npi_rpower_scalar)
112112
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::rpower>);
113113

114-
NNVM_REGISTER_OP(_npi_fmax)
115-
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastCompute<gpu, mshadow_op::fmax>);
116-
117-
NNVM_REGISTER_OP(_backward_npi_fmax)
118-
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastBackwardUseIn<gpu, mshadow_op::ge,
119-
mshadow_op::lt>);
120-
121-
NNVM_REGISTER_OP(_npi_fmax_scalar)
122-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::fmax>);
123-
124-
NNVM_REGISTER_OP(_backward_npi_fmax_scalar)
125-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::ge>);
126-
127-
NNVM_REGISTER_OP(_npi_fmin)
128-
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastCompute<gpu, mshadow_op::fmin>);
129-
130-
NNVM_REGISTER_OP(_backward_npi_fmin)
131-
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastBackwardUseIn<gpu, mshadow_op::le,
132-
mshadow_op::gt>);
133-
134-
NNVM_REGISTER_OP(_npi_fmin_scalar)
135-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::fmin>);
136-
137-
NNVM_REGISTER_OP(_backward_npi_fmin_scalar)
138-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::le>);
139-
140-
NNVM_REGISTER_OP(_npi_fmod)
141-
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastCompute<gpu, mshadow_op::fmod>);
142-
143-
NNVM_REGISTER_OP(_backward_npi_fmod)
144-
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastBackwardUseIn<gpu, mshadow_op::mod_grad,
145-
mshadow_op::mod_rgrad>);
146-
147-
NNVM_REGISTER_OP(_npi_fmod_scalar)
148-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::fmod>);
149-
150-
NNVM_REGISTER_OP(_backward_npi_fmod_scalar)
151-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::mod_grad>);
152-
153-
NNVM_REGISTER_OP(_npi_rfmod_scalar)
154-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::rfmod>);
155-
156-
NNVM_REGISTER_OP(_backward_npi_rfmod_scalar)
157-
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::rmod_grad>);
158-
159114
} // namespace op
160115
} // namespace mxnet

‎src/operator/numpy/np_elemwise_broadcast_op_extended_sec.cc‎

Lines changed: 142 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,142 @@
1+
/*
2+
* Licensed to the Apache Software Foundation (ASF) under one
3+
* or more contributor license agreements. See the NOTICE file
4+
* distributed with this work for additional information
5+
* regarding copyright ownership. The ASF licenses this file
6+
* to you under the Apache License, Version 2.0 (the
7+
* "License"); you may not use this file except in compliance
8+
* with the License. You may obtain a copy of the License at
9+
*
10+
* http://www.apache.org/licenses/LICENSE-2.0
11+
*
12+
* Unless required by applicable law or agreed to in writing,
13+
* software distributed under the License is distributed on an
14+
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15+
* KIND, either express or implied. See the License for the
16+
* specific language governing permissions and limitations
17+
* under the License.
18+
*/
19+
20+
/*!
21+
* Copyright (c) 2019 by Contributors
22+
* \file np_elemwise_broadcast_op_extended_sec.cc
23+
* \brief CPU Implementation of extended functions for elementwise numpy binary broadcast operator. (Second extended file)
24+
*/
25+
26+
#include "../../common/utils.h"
27+
#include "./np_elemwise_broadcast_op.h"
28+
29+
namespace mxnet {
30+
namespace op {
31+
32+
#define MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(name) \
33+
NNVM_REGISTER_OP(name) \
34+
.set_num_inputs(1) \
35+
.set_num_outputs(1) \
36+
.set_attr_parser([](NodeAttrs* attrs) { \
37+
attrs->parsed = std::stod(attrs->dict["scalar"]); \
38+
}) \
39+
.set_attr<mxnet::FInferShape>("FInferShape", ElemwiseShape<1, 1>) \
40+
.set_attr<nnvm::FInferType>("FInferType", NumpyBinaryScalarType) \
41+
.set_attr<nnvm::FInplaceOption>("FInplaceOption", \
42+
[](const NodeAttrs& attrs){ \
43+
return std::vector<std::pair<int, int> >{{0, 0}}; \
44+
}) \
45+
.add_argument("data", "NDArray-or-Symbol", "source input") \
46+
.add_argument("scalar", "float", "scalar input")
47+
48+
MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(_npi_fmax)
49+
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::fmax>)
50+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmax"});
51+
52+
NNVM_REGISTER_OP(_backward_npi_fmax)
53+
.set_num_inputs(3)
54+
.set_num_outputs(2)
55+
.set_attr<nnvm::TIsBackward>("TIsBackward", true)
56+
.set_attr<nnvm::FInplaceOption>("FInplaceOption",
57+
[](const NodeAttrs& attrs){
58+
return std::vector<std::pair<int, int> >{{0, 1}};
59+
})
60+
.set_attr<FResourceRequest>("FResourceRequest",
61+
[](const NodeAttrs& attrs) {
62+
return std::vector<ResourceRequest>{ResourceRequest::kTempSpace};
63+
})
64+
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastBackwardUseIn<cpu, mshadow_op::ge,
65+
mshadow_op::lt>);
66+
67+
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_fmax_scalar)
68+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::fmax>)
69+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmax_scalar"});
70+
71+
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_fmax_scalar)
72+
.add_argument("scalar", "float", "scalar value")
73+
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
74+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::ge>);
75+
76+
MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(_npi_fmin)
77+
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::fmin>)
78+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmin"});
79+
80+
NNVM_REGISTER_OP(_backward_npi_fmin)
81+
.set_num_inputs(3)
82+
.set_num_outputs(2)
83+
.set_attr<nnvm::TIsBackward>("TIsBackward", true)
84+
.set_attr<nnvm::FInplaceOption>("FInplaceOption",
85+
[](const NodeAttrs& attrs){
86+
return std::vector<std::pair<int, int> >{{0, 1}};
87+
})
88+
.set_attr<FResourceRequest>("FResourceRequest",
89+
[](const NodeAttrs& attrs) {
90+
return std::vector<ResourceRequest>{ResourceRequest::kTempSpace};
91+
})
92+
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastBackwardUseIn<cpu, mshadow_op::le,
93+
mshadow_op::gt>);
94+
95+
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_fmin_scalar)
96+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::fmin>)
97+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmin_scalar"});
98+
99+
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_fmin_scalar)
100+
.add_argument("scalar", "float", "scalar value")
101+
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
102+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::le>);
103+
104+
MXNET_OPERATOR_REGISTER_BINARY_BROADCAST(_npi_fmod)
105+
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastCompute<cpu, mshadow_op::fmod>)
106+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmod"});
107+
108+
NNVM_REGISTER_OP(_backward_npi_fmod)
109+
.set_num_inputs(3)
110+
.set_num_outputs(2)
111+
.set_attr<nnvm::TIsBackward>("TIsBackward", true)
112+
.set_attr<nnvm::FInplaceOption>("FInplaceOption",
113+
[](const NodeAttrs& attrs){
114+
return std::vector<std::pair<int, int> >{{0, 1}};
115+
})
116+
.set_attr<FResourceRequest>("FResourceRequest",
117+
[](const NodeAttrs& attrs) {
118+
return std::vector<ResourceRequest>{ResourceRequest::kTempSpace};
119+
})
120+
.set_attr<FCompute>("FCompute<cpu>", BinaryBroadcastBackwardUseIn<cpu, mshadow_op::mod_grad,
121+
mshadow_op::mod_rgrad>);
122+
123+
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_fmod_scalar)
124+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::fmod>)
125+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_fmod_scalar"});
126+
127+
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_fmod_scalar)
128+
.add_argument("scalar", "float", "scalar value")
129+
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
130+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::mod_grad>);
131+
132+
MXNET_OPERATOR_REGISTER_NP_BINARY_SCALAR(_npi_rfmod_scalar)
133+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Compute<cpu, mshadow_op::rfmod>)
134+
.set_attr<nnvm::FGradient>("FGradient", ElemwiseGradUseIn{"_backward_npi_rfmod_scalar"});
135+
136+
MXNET_OPERATOR_REGISTER_BINARY(_backward_npi_rfmod_scalar)
137+
.add_argument("scalar", "float", "scalar value")
138+
.set_attr_parser([](NodeAttrs *attrs) { attrs->parsed = std::stod(attrs->dict["scalar"]); })
139+
.set_attr<FCompute>("FCompute<cpu>", BinaryScalarOp::Backward<cpu, mshadow_op::rmod_grad>);
140+
141+
} // namespace op
142+
} // namespace mxnet

‎src/operator/numpy/np_elemwise_broadcast_op_extended_sec.cu‎

Lines changed: 77 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,77 @@
1+
/*
2+
* Licensed to the Apache Software Foundation (ASF) under one
3+
* or more contributor license agreements. See the NOTICE file
4+
* distributed with this work for additional information
5+
* regarding copyright ownership. The ASF licenses this file
6+
* to you under the Apache License, Version 2.0 (the
7+
* "License"); you may not use this file except in compliance
8+
* with the License. You may obtain a copy of the License at
9+
*
10+
* http://www.apache.org/licenses/LICENSE-2.0
11+
*
12+
* Unless required by applicable law or agreed to in writing,
13+
* software distributed under the License is distributed on an
14+
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15+
* KIND, either express or implied. See the License for the
16+
* specific language governing permissions and limitations
17+
* under the License.
18+
*/
19+
20+
/*!
21+
* Copyright (c) 2019 by Contributors
22+
* \file np_elemwise_broadcast_op_extended_sec.cu
23+
* \brief GPU Implementation of extended functions for elementwise binary broadcast operator. (Second extended file)
24+
*/
25+
26+
#include "./np_elemwise_broadcast_op.h"
27+
28+
namespace mxnet {
29+
namespace op {
30+
31+
NNVM_REGISTER_OP(_npi_fmax)
32+
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastCompute<gpu, mshadow_op::fmax>);
33+
34+
NNVM_REGISTER_OP(_backward_npi_fmax)
35+
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastBackwardUseIn<gpu, mshadow_op::ge,
36+
mshadow_op::lt>);
37+
38+
NNVM_REGISTER_OP(_npi_fmax_scalar)
39+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::fmax>);
40+
41+
NNVM_REGISTER_OP(_backward_npi_fmax_scalar)
42+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::ge>);
43+
44+
NNVM_REGISTER_OP(_npi_fmin)
45+
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastCompute<gpu, mshadow_op::fmin>);
46+
47+
NNVM_REGISTER_OP(_backward_npi_fmin)
48+
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastBackwardUseIn<gpu, mshadow_op::le,
49+
mshadow_op::gt>);
50+
51+
NNVM_REGISTER_OP(_npi_fmin_scalar)
52+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::fmin>);
53+
54+
NNVM_REGISTER_OP(_backward_npi_fmin_scalar)
55+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::le>);
56+
57+
NNVM_REGISTER_OP(_npi_fmod)
58+
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastCompute<gpu, mshadow_op::fmod>);
59+
60+
NNVM_REGISTER_OP(_backward_npi_fmod)
61+
.set_attr<FCompute>("FCompute<gpu>", BinaryBroadcastBackwardUseIn<gpu, mshadow_op::mod_grad,
62+
mshadow_op::mod_rgrad>);
63+
64+
NNVM_REGISTER_OP(_npi_fmod_scalar)
65+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::fmod>);
66+
67+
NNVM_REGISTER_OP(_backward_npi_fmod_scalar)
68+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::mod_grad>);
69+
70+
NNVM_REGISTER_OP(_npi_rfmod_scalar)
71+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Compute<gpu, mshadow_op::rfmod>);
72+
73+
NNVM_REGISTER_OP(_backward_npi_rfmod_scalar)
74+
.set_attr<FCompute>("FCompute<gpu>", BinaryScalarOp::Backward<gpu, mshadow_op::rmod_grad>);
75+
76+
} // namespace op
77+
} // namespace mxnet

0 commit comments

Comments
 (0)