Would love to help all implementations improve – would you be able to send Halide an email so we can take this offline? Happy to help if possible!