experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make kMaxBatchCount constexpr.

Browse Source 
Also define it to avoid link error. Resolves a TODO.

PiperOrigin-RevId: 360282145
Change-Id: I1f66bebeddcecc1dd59ad96cea04e59aef74cb21
This commit is contained in:
Reed Wanderman-Milne 2021-03-01 14:46:46 -08:00 committed by TensorFlower Gardener
parent d10071f208
commit 5d063e8c0e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/stream_executor/cuda/cuda_blas.cc
View File

@ -3338,9 +3338,7 @@ class CUDABlasLtMatmulPlan final : public blas::IBlasLtMatmulPlan {
private:
// In some cases cublasLt does not support large batch sizes, so we need to
// split up such cases into multiple calls.
// TODO(reedwm): Making this static or constexpr causes a link error with gcc
// in debug mode for unknown reasons. Investigate why.
const int kMaxBatchCount = 65535;
static constexpr int kMaxBatchCount = 65535;
blas::BlasLtMatmulPlanParams params_;
blas::DataType scale_type_;
UniqueOpDesc op_desc_;
@ -3358,6 +3356,8 @@ class CUDABlasLtMatmulPlan final : public blas::IBlasLtMatmulPlan {
UniqueLayoutDesc d_remainder_desc_;
};
/*static*/ constexpr int CUDABlasLtMatmulPlan::kMaxBatchCount;
bool CUDABlasLtMatmulPlan::SetBiasPointer(const void *bias) const {
return SetCublasLtAttr(op_desc_.get(), CUBLASLT_MATMUL_DESC_BIAS_POINTER,
bias)