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 2020/11/30 08:26:19 UTC

[GitHub] [tvm] codeislife99 opened a new pull request #6997: Maskrcnn cuda

codeislife99 opened a new pull request #6997:
URL: https://github.com/apache/tvm/pull/6997


   Thanks for contributing to TVM!   Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from [Reviewers](https://github.com/apache/incubator-tvm/blob/master/CONTRIBUTORS.md#reviewers) by @ them in the pull request thread.
   


----------------------------------------------------------------
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:
users@infra.apache.org

[GitHub] [tvm] codeislife99 closed pull request #6997: Maskrcnn cuda

Posted by GitBox <gi...@apache.org>.
codeislife99 closed pull request #6997:
URL: https://github.com/apache/tvm/pull/6997


   


----------------------------------------------------------------
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:
users@infra.apache.org

[GitHub] [tvm] trevor-m commented on a change in pull request #6997: Maskrcnn cuda

Posted by GitBox <gi...@apache.org>.
trevor-m commented on a change in pull request #6997:
URL: https://github.com/apache/tvm/pull/6997#discussion_r532784460



##########
File path: tests/python/contrib/test_tensorrt.py
##########
@@ -1034,5 +1037,121 @@ def set_func_attr(func, compile_name, symbol_name):
     tvm.ir.assert_structural_equal(mod_trt, mod_exp, map_free_vars=True)
 
 
+def test_maskrcnn_resnet50() -> None:
+    """
+    This function tests the working of pytorch maskrcnn with resnet50 as backbone with
+    VM and VM + TRT. Since the order of compiled model outputs is a bit different from
+    original pytorch model, it uses a custom logic for comparison check.
+    """
+    if skip_codegen_test() or skip_runtime_test():

Review comment:
       Move the `skip_runtime_test` after compilation




----------------------------------------------------------------
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:
users@infra.apache.org