You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2022/02/26 14:30:41 UTC

[GitHub] [tvm] JCBrouwer opened a new pull request #10396: Enable groups argument for conv2d_transpose on the cudnn backend

JCBrouwer opened a new pull request #10396:
URL: https://github.com/apache/tvm/pull/10396


   Some discussion here: #10223
   
   @masahi 


-- 
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 commented on a change in pull request #10396: Enable groups argument for conv2d_transpose on the cudnn backend

Posted by GitBox <gi...@apache.org>.
masahi commented on a change in pull request #10396:
URL: https://github.com/apache/tvm/pull/10396#discussion_r815345574



##########
File path: tests/python/frontend/pytorch/test_forward.py
##########
@@ -4198,4 +4205,5 @@ def forward(self, x):
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    # pytest.main([__file__])
+    test_forward_conv2d_transpose_group()

Review comment:
       Remove this diff




-- 
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 #10396: Enable groups argument for conv2d_transpose on the cudnn backend

Posted by GitBox <gi...@apache.org>.
masahi merged pull request #10396:
URL: https://github.com/apache/tvm/pull/10396


   


-- 
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 commented on a change in pull request #10396: Enable groups argument for conv2d_transpose on the cudnn backend

Posted by GitBox <gi...@apache.org>.
masahi commented on a change in pull request #10396:
URL: https://github.com/apache/tvm/pull/10396#discussion_r815345554



##########
File path: tests/python/frontend/pytorch/test_forward.py
##########
@@ -1090,7 +1092,12 @@ def forward(self, x, w, s):
     styles = torch.rand(b)
 
     # cuda not supported for group > 1 conv2d_transpose
-    verify_trace_model(ModulatedConvTranspose2D().eval(), [inputs, weights, styles], ["llvm"])
+    targets = []

Review comment:
       Add llvm here




-- 
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