[Llama 3.2 1B] Unsupported Ops in Self-Attention Test (repeat_interleave)

[mstojkovicTT]

Steps to Reproduce:

To reproduce the error, run the following command:
pytest -svv forge/test/mlir/llama/tests/test_llama_self_attn.py::test_llama_self_attn["meta-llama/Llama-3.2-1B"]

Expected Behavior:

The test should pass without any errors, verifying the correct behavior of the self-attention mechanism.

Actual Behavior:

The test fails during the RUN_MLIR_COMPILER stage with the following error:

Unsupported Ops at: RUN_MLIR_COMPILER stage
repeat_interleave
                 Input_shape: [{1, 8, 1, 12, 64}, ]
                                         Attributes: repeat_interleave(1,0,){dim: 0, repeats: 1}
                                         Attributes: repeat_interleave(4,2,){dim: 2, repeats: 4}

Aditional info

The repeat_interleave operation is currently unsupported. However, ttnn.repeat_interleave support is available in Metal and awaiting uplift.