How to assign address space when using transform.structured.promote

Hi, is that possible to assign address space for memref.alloc generated by transform.structured.promote?

Today, no.

You should be able to add such a functionality by setting the allocation callback supplied through LinalgPromotionOptions inside PromoteOp::applyToOne.

Thank you, I will try

In a recent patch, I added the support. You can find the examples in this test case.

 %2 = transform.structured.promote %1 { operands_to_promote = [0, 1], mapping = [#gpu.memory_space<workgroup>] }

1 Like