[GitHub] [tvm] tkonolige opened a new pull request #10021: [TOPI,CUDA] Don't enable cudnn conv2d kernel if is not supported

[GitBox <gi...@apache.org>]

tkonolige opened a new pull request #10021:
URL: https://github.com/apache/tvm/pull/10021


   Specifically, check that layout is not NCHW if datatype is int8.
   
   @mbrookhart


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] tkonolige commented on pull request #10021: [TOPI,CUDA] Don't enable cudnn conv2d kernel if is not supported

[GitBox <gi...@apache.org>]

tkonolige commented on pull request #10021:
URL: https://github.com/apache/tvm/pull/10021#issuecomment-1018822444


   I've updated this PR to not use `conv2d_cudnn` if any of the inputs are int8. Via talking with @masahi, the issue is that cudnn only supports int8 inputs with int8 output while we require int32 outputs.


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] masahi merged pull request #10021: [TOPI,CUDA] Don't enable cudnn conv2d kernel if is not supported

[GitBox <gi...@apache.org>]

masahi merged pull request #10021:
URL: https://github.com/apache/tvm/pull/10021


   


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org