search

pytorch / tvm

TVM integration into PyTorch
449 stars 64 forks source link

RuntimeError: _Map_base::at #148

Open lolongcovas opened 4 years ago

lolongcovas commented 4 years ago

python3 test/benchmarks.py

output:

Warming JIT up with 10 runs
Running JIT 100 times
Done benchmarking JIT
Tracing model with TVM
Warming TVM up with 10 iters
Traceback (most recent call last):

  File "test/benchmarks.py", line 82, in <module>
    run_benchmark(csv_file)

  File "test/benchmarks.py", line 75, in run_benchmark
    benchmark(model, csv_file)

  File "test/benchmarks.py", line 44, in benchmark
    _ = trace_tvm(*inputs)

  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 539, in __call__
    result = self.forward(*input, **kwargs)

RuntimeError: _Map_base::at
The above operation failed in interpreter.
Traceback (most recent call last):

packages:

torch: 1.4.0a0+6f62c31
llvm-config --version: 6.0.0