return func.getArgAttrOfType<mlir::StringAttr>(
          resource_arg.getArgNumber(), kFuncDeviceAttr);
    } else {
      return mlir::Attribute{};
    }
  }
}

bool IsCompositeDevice(mlir::Attribute attr) {
  const auto str_attr = attr.dyn_cast_or_null<mlir::StringAttr>();
  return str_attr &&
         (str_attr.getValue().find("COMPOSITE") != llvm::StringRef::npos);
}
}  // namespace