check cudnn version (#1175)

cf3e2992 · Fenglei · Robert Kimball · 846f6bfe · cf3e2992
Commit cf3e2992 authored Jun 28, 2018 by Fenglei Committed by Robert Kimball Jun 28, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

cudnn_host_parameters.hpp src/ngraph/runtime/gpu/cudnn_host_parameters.hpp +2 -0

No files found.
--- a/src/ngraph/runtime/gpu/cudnn_host_parameters.hpp
+++ b/src/ngraph/runtime/gpu/cudnn_host_parameters.hpp
@@ -60,8 +60,10 @@ namespace ngraph
                        break;
                    case CUDNN_DATA_HALF:
                    case CUDNN_DATA_INT8x4:
+#if CUDNN_VERSION >= 7100
                    case CUDNN_DATA_UINT8:
                    case CUDNN_DATA_UINT8x4:
+#endif
                        std::string err = "datatype is not supported by cuDNN";
                        throw std::runtime_error(err);
                    }