iefgnoix

Lower randperm.generator_out

1

Per this Github issue https://github.com/pytorch/xla/issues/3369, this PR intends to lower `randperm.generator_out`.

### System Info Hi. I'm using huggingface model `bert base` on cloud TPU but it couldn't generate the `--logging_dir` that I expected. ### Who can help? _No response_ ### Information...

Make it convenient for users so as not having to install vim every time.

Add XLA unit tests to pre-submit CI

13

### System Info Hi team, Can we add some unit test for XLA (GPU, TPU v4, TPU v2 or v3) to the pre-submit CI? The test can be as simple...

Add missing type uint16, uint32, and uint64 to TensorHash in LTC.

1

If I do: ``` xla_device = xm.xla_device() xla_tensor_0 = torch.tensor(42, dtype=torch.uint32).to(xla_device) ``` I got the error: ``` RuntimeError: false INTERNAL ASSERT FAILED at "/ansible/pytorch/torch/csrc/lazy/core/hash.h":139, please report a bug to PyTorch....

# Description Please include a summary of relevant context/issue and your changes. # Tests Please describe the tests that you ran on TPUs to verify changes. **Instruction and/or command lines...

Not implemented: Non-trivial layouts unsupported

1

### Description Hi. I am extending the Pallas paged attention kernel. The case is a MQA. When I run my kernel, I encountered the following error which suggests it is...