[mosaic] Enabled `use-nameloc-as-prefix` for IR dumps
At the moment the named locations use JAX's primtive names. So the IR looks like
```
%run_scoped3A = memref.alloca() : memref<2x8xi32, #tpu.memory_space<vmem>>
%run_scoped3A_0 = memref.alloca() : memref<2x8xi32, #tpu.memory_space<vmem>>
%select_n3A = arith.constant true
```
PiperOrigin-RevId: 938631137
