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 #include "tensorflow/stream_executor/cuda/cudnn_version.h" 17 18 namespace stream_executor { 19 namespace gpu { 20 IsSourceCompatibleWithCudnnLibrary(CudnnVersion source_version,CudnnVersion loaded_version)21bool IsSourceCompatibleWithCudnnLibrary(CudnnVersion source_version, 22 CudnnVersion loaded_version) { 23 // Major version is neither forward or backward compatible and therefore major 24 // versions needs to match between source and library. 25 // 26 // Minor version is backward-compatible beginning with CuDNN 7 and therefore 27 // minor version of library needs to be same or higher. 28 // 29 // Patch releases are always forward and backward compatible and therefore 30 // need not match. 31 if (loaded_version.major_version != source_version.major_version) { 32 return false; 33 } 34 return ((loaded_version.minor_version == source_version.minor_version) || 35 (source_version.major_version >= 7 && 36 loaded_version.minor_version >= source_version.minor_version)); 37 } 38 39 } // namespace gpu 40 } // namespace stream_executor 41