jwfromm opened a new pull request #5376: [Relay][Strategy] Add cuda target 
check to dense tensorcore schedule.
URL: https://github.com/apache/incubator-tvm/pull/5376
 
 
   Calling `nvcc.have_tensorcore` will cause errors when using a non-nvidia 
GPU. This one line PR adds a check that the current target is `cuda` before the 
`nvcc` call. The equivalent `conv2d` and `conv3d` strategies already have this 
check, but it looks like we forgot to add it to `dense`. Hopefully, this 
addresses the error found in Issue #5370.
   

----------------------------------------------------------------
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.
 
For queries about this service, please contact Infrastructure at:
[email protected]


With regards,
Apache Git Services

Reply via email to