FIX some OMPT tests that fail on RISC-V (#156914)

There is a known issue with these tests (see
https://github.com/llvm/llvm-project/issues/35661).
It looks that the behaviour is very architecture specific and it was
somehow solved for aarch64 but no mention to riscv64.

GitOrigin-RevId: 4e19543b1a85e635a8c663081bfdf42488795623