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

[MXNET-507] Set dtype=int32 for ret_indices in ordering ops #11134

Closed
wants to merge 18 commits into from
Closed

[MXNET-507] Set dtype=int32 for ret_indices in ordering ops #11134

Show file tree
Hide file tree
Changes from 16 commits
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
226aee3
set the dtype of index to be int32 for ordering ops
sxjscience Jun 4, 2018
0781f55
fix
sxjscience Jun 4, 2018
9c22358
fix
sxjscience Jun 4, 2018
d38cc01
fix
sxjscience Jun 4, 2018
5433b97
Merge remote-tracking branch 'apache/master' into fix_order
sxjscience Jun 11, 2018
7b771a0
Merge remote-tracking branch 'upstream/master' into fix_order
sxjscience Jun 12, 2018
4bb6be7
try to add the dtype option
sxjscience Jun 12, 2018
d16f614
fix
sxjscience Jun 12, 2018
9d3bdf3
try to fix
sxjscience Jun 12, 2018
3c1d162
try to fix
sxjscience Jun 12, 2018
51736e1
fix lint
sxjscience Jun 12, 2018
c1016c7
add more tests
sxjscience Jun 12, 2018
b26d24b
fix test
sxjscience Jun 12, 2018
ad0fb51
Merge remote-tracking branch 'upstream/master' into fix_order
sxjscience Jun 15, 2018
cde4b5f
Do not change unrelated file
sxjscience Jun 15, 2018
1f805d9
Merge remote-tracking branch 'upstream/master' into fix_order
sxjscience Jun 19, 2018
30a4457
remove float16
sxjscience Jul 4, 2018
c4bae4d
Revert "remove float16"
sxjscience Jul 4, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 45 additions & 0 deletions src/operator/mxnet_op.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -176,6 +176,51 @@ inline int get_num_threads<cpu>(const int N) {
LOG(FATAL) << "Unknown type enum " << type; \
}

#define MXNET_NO_FLOAT16_TYPE_SWITCH(type, DType, ...) \
switch (type) { \
case mshadow::kFloat32: \
{ \
typedef float DType; \
{__VA_ARGS__} \
} \
break; \
case mshadow::kFloat64: \
{ \
typedef double DType; \
{__VA_ARGS__} \
} \
break; \
case mshadow::kFloat16: \
LOG(FATAL) << "This operation does not " \
"support float16"; \
break; \
case mshadow::kUint8: \
{ \
typedef uint8_t DType; \
{__VA_ARGS__} \
} \
break; \
case mshadow::kInt8: \
{ \
typedef int8_t DType; \
{__VA_ARGS__} \
} \
break; \
case mshadow::kInt32: \
{ \
typedef int32_t DType; \
{__VA_ARGS__} \
} \
break; \
case mshadow::kInt64: \
{ \
typedef int64_t DType; \
{__VA_ARGS__} \
} \
break; \
default: \
LOG(FATAL) << "Unknown type enum " << type; \
}

/*!
* \brief assign the val to out according
Expand Down
Loading