@@ -91,33 +91,33 @@ struct Benchmark : public ::testing::Test {
91
91
void SetUp () {
92
92
if (!FLAGS_disable_version_checks) {
93
93
auto cudnnVersion = cudnnGetVersion ();
94
- CHECK_GE (6021 , cudnnVersion)
94
+ CHECK_LE (6021 , cudnnVersion)
95
95
<< " [CUDNN][VERSION] Enforce version compatibility check" ;
96
96
97
97
auto cudaRtVersion = cudnnGetCudartVersion ();
98
- CHECK_GE (8000 , cudaRtVersion)
98
+ CHECK_LE (8000 , cudaRtVersion)
99
99
<< " [CUDART][VERSION] Enforce version compatibility check" ;
100
100
101
101
int cublasVersion;
102
102
cublasHandle_t handle;
103
103
TC_CUDA_CUBLAS_ENFORCE (cublasCreate_v2 (&handle));
104
104
TC_CUDA_CUBLAS_ENFORCE (cublasGetVersion_v2 (handle, &cublasVersion));
105
- CHECK_GE (8000 , cublasVersion)
105
+ CHECK_LE (8000 , cublasVersion)
106
106
<< " [CUBLAS][VERSION] Enforce version compatibility check" ;
107
107
tc::ScopeGuard sg (
108
108
[&handle]() { TC_CUDA_CUBLAS_ENFORCE (cublasDestroy_v2 (handle)); });
109
109
110
110
int cudaRuntimeVersion;
111
111
TC_CUDA_RUNTIMEAPI_ENFORCE (cudaRuntimeGetVersion (&cudaRuntimeVersion));
112
- CHECK_GE (8000 , cudaRuntimeVersion)
112
+ CHECK_LE (8000 , cudaRuntimeVersion)
113
113
<< " [CUDA RUNTIME][VERSION] Enforce version compatibility check" ;
114
114
115
115
int nvrtcVersionMajor;
116
116
int nvrtcVersionMinor;
117
117
TC_NVRTC_CHECK (nvrtcVersion (&nvrtcVersionMajor, &nvrtcVersionMinor));
118
- CHECK_GE (8 , nvrtcVersionMajor)
118
+ CHECK_LE (8 , nvrtcVersionMajor)
119
119
<< " [NVRTC][MAJOR][VERSION] Enforce version compatibility check" ;
120
- CHECK_GE (0 , nvrtcVersionMinor)
120
+ CHECK_LE (0 , nvrtcVersionMinor)
121
121
<< " [NVRTC][MINOR][VERSION] Enforce version compatibility check" ;
122
122
}
123
123
}
0 commit comments