@kvegiraj @srkreddy1238
I was able to fix this issue by doing the transform and clml partitioning under tvm.transform.PassContext(opt_level=3) optimization pass.
with tvm.transform.PassContext(opt_level=3):
mod = seq(mod)
print("----------ir after layout change------------")
print(mod)
print("----------------------")
if not local_demo and enable_clml:
print("partition clml")
print(clml.is_clml_runtime_enabled())
mod = clml.preprocess_module(mod)
mod = clml.partition_for_clml(mod, params)
target = tvm.target.Target(test_target, host=target)
#mod = seq(mod)
lib = relay.build(mod, target=target, params=params)