From 934fde5b8c60987db36438ab4f70f8a91bce306b Mon Sep 17 00:00:00 2001
From: James Qin <jamesqin@google.com>
Date: Mon, 8 Oct 2018 17:40:07 -0700
Subject: Register int64 SUM GPU kernel.

PiperOrigin-RevId: 216280913
---
 tensorflow/core/kernels/reduction_ops_sum.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/core/kernels/reduction_ops_sum.cc b/tensorflow/core/kernels/reduction_ops_sum.cc
index 5318d8c133..cf0d0f5c71 100644
--- a/tensorflow/core/kernels/reduction_ops_sum.cc
+++ b/tensorflow/core/kernels/reduction_ops_sum.cc
@@ -51,6 +51,7 @@ TF_CALL_NUMBER_TYPES(REGISTER_CPU_KERNELS);
           .HostMemory("reduction_indices"),                                    \
       ReductionOp<GPUDevice, type, int64, Eigen::internal::SumReducer<type>>);
 TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU_KERNELS);
+TF_CALL_int64(REGISTER_GPU_KERNELS);
 TF_CALL_complex64(REGISTER_GPU_KERNELS);
 TF_CALL_complex128(REGISTER_GPU_KERNELS);
 #undef REGISTER_GPU_KERNELS
-- 
cgit v1.2.3