| 24 | namespace gpu { |
| 25 | |
| 26 | struct CudnnVersion { |
| 27 | CudnnVersion() = default; |
| 28 | |
| 29 | CudnnVersion(int major, int minor, int patch) |
| 30 | : major_version(major), minor_version(minor), patch_level(patch) {} |
| 31 | |
| 32 | std::string ToString() const { |
| 33 | return absl::StrCat(major_version, ".", minor_version, ".", patch_level); |
| 34 | } |
| 35 | |
| 36 | int major_version; |
| 37 | int minor_version; |
| 38 | int patch_level; |
| 39 | }; |
| 40 | |
| 41 | // Returns true if the given source CuDNN version is compatible with the given |
| 42 | // loaded version. |