iree: resnet_fake_weights.mlir doesn't work in Linalg on tensors path

Describe the bug The issue happens in running the module. The compilation was done without any error messages.

The error message is from hal_module.c:

iree/modules/hal/hal_module.c:742: OUT_OF_RANGE; binding count 37 > 32; while invoking native function hal.descriptor_set_layout.create; while calling import; Creating context; Evaluating functions

https://github.com/google/iree/blob/053c1a4bda165ec24cdd43b11bc2dec2a9a893c6/iree/modules/hal/hal_module.c#L738-L743

To Reproduce

Run: iree-run-mlir iree/test/e2e/models/resnet_fake_weights.mlir -iree-hal-target-backends=dylib-llvm-aot -function-input="1x224x224x3xf32"

You can also use iree-translate + iree-run-module. The issue will happen in iree-run-module.

About this issue

  • Original URL
  • State: closed
  • Created 3 years ago
  • Comments: 15 (8 by maintainers)

Commits related to this issue

Most upvoted comments

The whole log file is too big, I have the IR after OrdinalAllocationPass https://gist.github.com/hanhanW/5c6badfa299ec757fa3e38f45aad753c

The whole log file can be downloaded from https://drive.google.com/file/d/1Tb8_ANEpS3NFFDjRN9bEnosabBHpgeNL/view?usp=sharing