Thank you @srkreddy1238.
Also facing some unsupported operator error at opt_level=1 for both CUDA and OPENCL targets.
Below is the Error:
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 132, in schedule_reduce
traverse_after_reduce(outs[0].op)
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 115, in traverse_after_reduce
traverse_after_reduce(tensor.op)
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 120, in traverse_after_reduce
traverse_before_reduce(tensor.op)
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 103, in traverse_before_reduce
traverse_before_reduce(tensor.op)
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 103, in traverse_before_reduce
traverse_before_reduce(tensor.op)
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 103, in traverse_before_reduce
traverse_before_reduce(tensor.op)
[Previous line repeated 4 more times]
File "/home/ubuntu/.local/lib/python3.6/site-packages/topi-0.6.dev0-py3.6.egg/topi/cuda/reduction.py", line 105, in traverse_before_reduce
raise RuntimeError("Unsupported operator: %s" % operator.tag)
RuntimeError: Unsupported operator:
Error during compile func
Any help here would be great.
Thank you