diff --git a/src/relay/transforms/annotate_texture_storage.cc b/src/relay/transforms/annotate_texture_storage.cc index a58a603d1639..a7eca1ebe70c 100644 --- a/src/relay/transforms/annotate_texture_storage.cc +++ b/src/relay/transforms/annotate_texture_storage.cc @@ -326,7 +326,7 @@ class StorageInfo : private transform::DeviceAwareExprVisitor { } } else if (auto attrs = call->attrs.as()) { if ((attrs->data_layout == "NCHW4c" || attrs->data_layout == "NHWC4c") && - (attrs->kernel_layout == "OIHW4o" || attrs->kernel_layout == "HWIO4o")) { + (attrs->kernel_layout == "OIHW4o" || attrs->kernel_layout == "HWIO4o")) { supports_texture_storage = true; } } else if (auto attrs = call->attrs.as()) {