search

cornell-zhang / heterocl

HeteroCL: A Multi-Paradigm Programming Infrastructure for Software-Defined Heterogeneous Computing
https://cornell-zhang.github.io/heterocl/
Apache License 2.0
322 stars 92 forks source link

Modified BNN batch norm function (linear equation) breaking TVM stage #300

Closed paldebjit closed 1 year ago

paldebjit commented 3 years ago 
(hcl) [dp638@zhang-24 bnn]$ python resnet_aocl.py 
Loading the data.
Files already downloaded and verified
[15:30:06] Mark stage conv1_pad on FPGA scope...
[15:30:06] Mark stage conv1 on FPGA scope...
[15:30:06] Mark stage bn1 on FPGA scope...
[15:30:06] Mark stage bn1 on FPGA scope...
Traceback (most recent call last):
  File "resnet_aocl.py", line 11, in <module>
    resnet20 = build_resnet20_inf(params, target)
  File "/scratch/users/dp638/Work/heterocl/samples/bnn/resnet_main.py", line 169, in build_resnet20_inf
    s.to([input_image] + hcl_ph, target.xcel)
  File "/home/dp638/Work/CondaEnvs/hcl/lib/python3.7/site-packages/heterocl-0.3-py3.7.egg/heterocl/schedule.py", line 359, in to
    ret = self.sch.to(target, dst, src, axis, mode, depth, burst_len)
  File "/home/dp638/Work/CondaEnvs/hcl/lib/python3.7/site-packages/heterocl-0.3-py3.7.egg/heterocl/tvm/schedule.py", line 394, in to
    io_type, depth, dev_port)
  File "/home/dp638/Work/CondaEnvs/hcl/lib/python3.7/site-packages/heterocl-0.3-py3.7.egg/heterocl/tvm/_ffi/function.py", line 280, in my_api_func
    return flocal(*args)
  File "/home/dp638/Work/CondaEnvs/hcl/lib/python3.7/site-packages/heterocl-0.3-py3.7.egg/heterocl/tvm/_ffi/_ctypes/function.py", line 183, in __call__
    ctypes.byref(ret_val), ctypes.byref(ret_tcode)))
  File "/home/dp638/Work/CondaEnvs/hcl/lib/python3.7/site-packages/heterocl-0.3-py3.7.egg/heterocl/tvm/_ffi/base.py", line 66, in check_call
    raise TVMError(py_str(_LIB.TVMGetLastError()))
heterocl.tvm._ffi.base.TVMError: [15:30:06] src/schedule/schedule_lang.cc:661: Check failed: it != (*this)->stage_map.end() Cannot find Stage for operator placeholder(bn1_running_mean, 0x555872f16a28) in the schedule

298

Modified nn code for BNN is attached. nn_.zip

(Look around line 740)

chhzh123 commented 1 year ago

As we have released a new version of HeteroCL that entirely moves the compilation stack to MLIR, issues related to the original HalideIR flow will be closed.