1 /* Copyright 2018 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 16 #ifndef TENSORFLOW_STREAM_EXECUTOR_CUDA_CUDNN_VERSION_H_ 17 #define TENSORFLOW_STREAM_EXECUTOR_CUDA_CUDNN_VERSION_H_ 18 19 #include <string> 20 21 #include "tensorflow/core/lib/strings/strcat.h" 22 23 namespace stream_executor { 24 namespace gpu { 25 26 struct CudnnVersion { 27 CudnnVersion() = default; 28 CudnnVersionCudnnVersion29 CudnnVersion(int major, int minor, int patch) 30 : major_version(major), minor_version(minor), patch_level(patch) {} 31 ToStringCudnnVersion32 tensorflow::string ToString() const { 33 return tensorflow::strings::StrCat(major_version, ".", minor_version, ".", 34 patch_level); 35 } 36 37 int major_version; 38 int minor_version; 39 int patch_level; 40 }; 41 42 // Returns true if the given source CuDNN version is compatible with the given 43 // loaded version. 44 bool IsSourceCompatibleWithCudnnLibrary(CudnnVersion source_version, 45 CudnnVersion loaded_version); 46 47 } // namespace gpu 48 } // namespace stream_executor 49 50 #endif // TENSORFLOW_STREAM_EXECUTOR_CUDA_CUDNN_VERSION_H_ 51