Add helper functions getDefaultXLAGenerator and createXLAGenerator to XLA random number generator

[iwknow]

Add helper functions getDefaultXLAGenerator and createXLAGenerator to XLA random number generator

These helper functions will be used with XLA hook later.

Refer to #9159

[iwknow]

@qihqi do you know why do i always get build timeout? it builds successfully locally on my machine. i am not able to find any clue of the cause from the build log. please take a look

[iwknow]

@ysiraichi @qihqi can you please take a look. thanks!

[ysiraichi]

do you know why do i always get build timeout?

This might be because your PR is not from a branch on this PyTorch/XLA repository.
That causes the CI not to use the remote cache.
I'm currently working on using GitHub cache for mitigating that #9659.

[ysiraichi]

Thank you for the PR. I think that it looks good overall.
Could you add a few C++ tests and check everything is working?

[iwknow]

strangely, i am no longer able to build the //test/cpp:test_xla_generator. The command that i use bazel build //test/cpp:test_xla_generator --experimental_ui_max_stdouterr_bytes=-1 the last flag is to let it print without length limit. the error i get is:

bazel-out/k8-opt/bin/_solib_k8/_U_A_Atorch_S_S_Clibc10___Ubuild_Slib/libc10.so: error: undefined reference to 'log', version 'GLIBC_2.29'
/usr/local/lib/libpython3.10.so: error: undefined reference to 'sem_clockwait', version 'GLIBC_2.30'
bazel-out/k8-opt/bin/_solib_k8/_U_A_Atorch_S_S_Clibc10___Ubuild_Slib/libc10.so: error: undefined reference to 'pthread_cond_clockwait', version 'GLIBC_2.30'
collect2: error: ld returned 1 exit status
Target //test/cpp:test_xla_generator failed to build

@ysiraichi do you have any clue about the issue? i was previously able to build and run //test/cpp:test_xla_generator

[ysiraichi]

Not sure what happened there.
It looks like you have an old glibc version in your system.
Are you using the docker image?

[iwknow]

Not sure what happened there. It looks like you have an old glibc version in your system. Are you using the docker image?

i am using the tpu-contributor dev container. I got the following when i run ldd --version

ldd (Debian GLIBC 2.31-13+deb11u11) 2.31
Copyright (C) 2020 Free Software Foundation, Inc.
This is free software; see the source for copying conditions.  There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
Written by Roland McGrath and Ulrich Drepper.

could this be related to build cache? or it is incompatible with my pytorch? should i sync my local pytorch version to the head? Also, i don't have local pytorch-xla installed as a python package. i am not sure if this is related.

[ysiraichi]

Try recompiling everything from scratch: PyTorch and PyTorch/XLA (clean the cache).