co-authored by
Aniket Shivam
parent
ca23ff7924
commit
b72cbf957d
@@ -603,7 +603,7 @@ int main(int argc, char const **args) {
|
||||
cudaDeviceProp props;
|
||||
CUDA_CHECK(cudaGetDeviceProperties(&props, 0));
|
||||
|
||||
if (!(props.major > 8 || (props.major == 8 && props.minor >= 0))) {
|
||||
if (!(props.major >= 8)) {
|
||||
std::cerr << "Ampere Tensor Ops must be run on a machine with compute capability at least 80."
|
||||
<< std::endl;
|
||||
notSupported = true;
|
||||
|
||||
Reference in New Issue
Block a user