[Apache TVM Discuss] [Questions] Run test_cutlass.py error:unexpected keyword argument 'capture_output'

2021-11-03 Thread masahi via Apache TVM Discuss
Oh right, I forgot to fix that. This happens on Python 3.6 or older, so can you try 3.7 or 3.8? --- [Visit Topic](https://discuss.tvm.apache.org/t/run-test-cutlass-py-error-unexpected-keyword-argument-capture-output/11400/2) to respond. You are receiving this because you enabled mailing

[Apache TVM Discuss] [Questions] Print auto_schedule Python schedule with topi op

2021-11-03 Thread Cody H. Yu via Apache TVM Discuss
Like I mentioned op name is defined in the TOPI compute, so auto-scheduler cannot change it. To solve this issue, we need to introduce a general utility to rename op names in a TE compute. --- [Visit Topic](https://discuss.tvm.apache.org/t/print-auto-schedule-python-schedule-with-topi-op/

[Apache TVM Discuss] [Questions] Run test_cutlass.py error:unexpected keyword argument 'capture_output'

2021-11-03 Thread ading via Apache TVM Discuss
When I run tests/python/contrib/test_cutlass.py, error: FAILED ../tests/python/contrib/test_cutlass.py::test_dense - TypeError: __init__() got an unexpected keyword argument 'capture_output' FAILED ../tests/python/contrib/test_cutlass.py::test_dense_bias - TypeError: __init__() got an unexpec

[Apache TVM Discuss] [Questions] Print auto_schedule Python schedule with topi op

2021-11-03 Thread Sergesg via Apache TVM Discuss
Perhaps adding a renaming logic [here](https://github.com/apache/tvm/blob/main/src/auto_scheduler/compute_dag.cc#L1206) may not work because there are just the print of `... = tuple(name.op.axis) + tuple(name.op.reduce_axis)`, but the following steps in [step print](https://github.com/apache/

[Apache TVM Discuss] [Questions] TOpPattern has not been registered for nn.dropout

2021-11-03 Thread Andrew Zhao Luo via Apache TVM Discuss
Yes that is correct. Though I believe someone was planning to work on this one in the next week. --- [Visit Topic](https://discuss.tvm.apache.org/t/toppattern-has-not-been-registered-for-nn-dropout/11305/9) to respond. You are receiving this because you enabled mailing list mode. To uns

[Apache TVM Discuss] [Questions] TOpPattern has not been registered for nn.dropout

2021-11-03 Thread Ligeng Zhu via Apache TVM Discuss
Hi Andrew, One more question, is similar behavior also made for the bn layer? Says it was fused to conv layer and does not have an actual implementation. --- [Visit Topic](https://discuss.tvm.apache.org/t/toppattern-has-not-been-registered-for-nn-dropout/11305/8) to respond. You are rec

[Apache TVM Discuss] [Questions] Relay failed to build models exported from pytorch

2021-11-03 Thread Ligeng Zhu via Apache TVM Discuss
Notice one very interesting one that might be helpful for this bug I tried to compile the models exported from mxnet ``` from tvm import relay import tvm import numpy as np import torch import torch as th import torch.nn as nn from torchvision import models import torch.onnx from tvm import

[Apache TVM Discuss] [Questions] Relay failed to build models exported from pytorch

2021-11-03 Thread Ligeng Zhu via Apache TVM Discuss
[quote="lhutton1, post:3, topic:11394"] elegant [/quote] Oh! Thanks for the information. You saved my day. --- [Visit Topic](https://discuss.tvm.apache.org/t/relay-failed-to-build-models-exported-from-pytorch/11394/5) to respond. You are receiving this because you enabled mailing list mo

[Apache TVM Discuss] [Questions] Relay failed to build models exported from pytorch

2021-11-03 Thread Luke Hutton via Apache TVM Discuss
Hi @Lyken17, I also ran into this issue recently. It turned out to be conflicting symbols between PyTorch and TVM, see https://github.com/apache/tvm/issues/9362#issuecomment-955263494 for the resolution. Alternatively, a quicker (but less elegant) solution is to import `torch` before `tvm`.

[Apache TVM Discuss] [Questions] Relay failed to build models exported from pytorch

2021-11-03 Thread Ligeng Zhu via Apache TVM Discuss
My environemnt ubuntu 20.04 | gcc: 9.3 | llvm: 10.0 | nvcc: 11.1 --- [Visit Topic](https://discuss.tvm.apache.org/t/relay-failed-to-build-models-exported-from-pytorch/11394/2) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [cl

[Apache TVM Discuss] [Questions] Relay failed to build models exported from pytorch

2021-11-03 Thread Ligeng Zhu via Apache TVM Discuss
Hi there, While I was following the tutorial [relay quick start](https://tvm.apache.org/docs/tutorial/relay_quick_start.html), I tried to load a module from pytorch but it raises segmentation fault error. The TVM I am using the latest commit `bff98843bef9a312587aaff51b679d9b69a7d5a7` and the

[Apache TVM Discuss] [Questions] LSTM mapped to specific op in custom target

2021-11-03 Thread Sameekbanerjee via Apache TVM Discuss
Is there a way to mark a layer as an LSTM layer after conversion to relay? I see almost RNN ops as broken into individual cells for each operator. I have a custom accelerator with First class support for LSTM at the operator level. Is there an easy way to do this ? Thanks in advance. --