diff options
| -rw-r--r-- | tensorflow/contrib/boosted_trees/kernels/model_ops.cc | 1 | ||||
| -rw-r--r-- | tensorflow/core/kernels/cuda_solvers.cc | 9 |
2 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/contrib/boosted_trees/kernels/model_ops.cc b/tensorflow/contrib/boosted_trees/kernels/model_ops.cc index d63be3d041..4b5d5ba0de 100644 --- a/tensorflow/contrib/boosted_trees/kernels/model_ops.cc +++ b/tensorflow/contrib/boosted_trees/kernels/model_ops.cc @@ -15,7 +15,6 @@ #include <string> #include "tensorflow/contrib/boosted_trees/lib/utils/tensor_utils.h" -#include "tensorflow/contrib/boosted_trees/proto/tree_config.pb.h" #include "tensorflow/contrib/boosted_trees/resources/decision_tree_ensemble_resource.h" #include "tensorflow/core/framework/op_kernel.h" #include "tensorflow/core/framework/resource_mgr.h" diff --git a/tensorflow/core/kernels/cuda_solvers.cc b/tensorflow/core/kernels/cuda_solvers.cc index 6c12a0e218..a83671a471 100644 --- a/tensorflow/core/kernels/cuda_solvers.cc +++ b/tensorflow/core/kernels/cuda_solvers.cc @@ -617,10 +617,11 @@ static inline Status GesvdImpl( // kernel on the stream, it is not a big performance hit. mutex_lock lock(handle_map_mutex); /* Launch the solver kernel. */ - TF_RETURN_IF_CUSOLVER_ERROR(solver( - cusolver_dn_handle, jobu, jobvt, m, n, CUDAComplex(A), lda, S, - CUDAComplex(U), ldu, CUDAComplex(VT), ldvt, - CUDAComplex(dev_workspace.mutable_data()), lwork, NULL, dev_lapack_info)); + TF_RETURN_IF_CUSOLVER_ERROR(solver(cusolver_dn_handle, jobu, jobvt, m, n, + CUDAComplex(A), lda, S, CUDAComplex(U), + ldu, CUDAComplex(VT), ldvt, + CUDAComplex(dev_workspace.mutable_data()), + lwork, nullptr, dev_lapack_info)); return Status::OK(); } |