Skip to content
This repository has been archived by the owner on Aug 11, 2020. It is now read-only.

Add compile flag to choose data type for tensor size due to performance degradation #371

Merged
merged 13 commits into from
Apr 12, 2019
Merged

Add compile flag to choose data type for tensor size due to performance degradation #371

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
4305465
add a compiler flag to select int64 type
apeforest Mar 29, 2019
44a5348
fix typo
apeforest Mar 29, 2019
0ea7165
fix compilation error
apeforest Mar 29, 2019
0c3a6e7
Merge remote-tracking branch 'upstream/master' into perf/large-tensor
apeforest Mar 29, 2019
359959e
add a compiler flag to select int64 type
apeforest Mar 29, 2019
6945913
fix typo
apeforest Mar 29, 2019
f618c0b
fix compilation error
apeforest Mar 29, 2019
12464b6
Merge branch 'perf/large-tensor' of /~https://github.com/apeforest/msha…
apeforest Apr 2, 2019
655a992
fix type in gemm functions
apeforest Apr 2, 2019
a0c5721
fix lint
apeforest Apr 2, 2019
8b28974
revert type change in gemm function call
apeforest Apr 9, 2019
bb6a037
sync change with upstream
apeforest Apr 9, 2019
9057bf9
remove uint32
apeforest Apr 11, 2019
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
6 changes: 5 additions & 1 deletion mshadow/base.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -285,7 +285,11 @@ const unsigned kRandBufferSize = 1000000;
/*! \brief pi */
const float kPi = 3.1415926f;
/*! \brief type that will be used for index */
typedef int64_t index_t;
#if MSHADOW_INT64_TENSOR_SIZE == 1
typedef int64_t index_t;
#else
typedef int32_t index_t;
#endif

#ifdef _WIN32
/*! \brief openmp index for windows */
Expand Down
4 changes: 2 additions & 2 deletions mshadow/tensor.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1069,10 +1069,10 @@ inline void BatchGEMM(Tensor<Device, 3, DType> dst,
#define MSHADOW_SCALAR_ double
#include "./expr_scalar-inl.h"
#undef MSHADOW_SCALAR_
#define MSHADOW_SCALAR_ int
#define MSHADOW_SCALAR_ int32_t
#include "./expr_scalar-inl.h"
#undef MSHADOW_SCALAR_
#define MSHADOW_SCALAR_ mshadow::index_t
#define MSHADOW_SCALAR_ int64_t
#include "./expr_scalar-inl.h"
#undef MSHADOW_SCALAR_
#define MSHADOW_SCALAR_ mshadow::half::half_t
Expand Down