Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add configuration for bazel #1264

Closed
wants to merge 1 commit into from
Closed

Conversation

Steboss
Copy link
Contributor

@Steboss Steboss commented Jan 22, 2025

This PR adds bazel's configuration step directly in the JAX container, so developers won't have to run it.
In particular, this is to avoid having devs experiencing problems such as:

RROR: /opt/xla/xla/service/gpu/kernels/BUILD:170:19: Compiling xla/service/gpu/kernels/topk_kernel_float.cu.cc failed: (Exit 1): crosstool_wrapper_driver_is_not_gcc failed: error executing command (from target //xla/service/gpu/kernels:topk_kernel_gpu_cuda) external/local_config_cuda/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc -MD -MF bazel-out/k8-opt/bin/xla/service/gpu/kernels/_objs/topk_kernel_gpu_cuda/topk_kernel_float.cu.d ... (remaining 207 arguments skipped)
/root/.cache/bazel/_bazel_root/172c56c881f24dddf5d15d4e92df59d7/execroot/xla/external/local_config_cuda/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc:227: SyntaxWarning: invalid escape sequence '\.'
  re.search('\.cpp$|\.cc$|\.c$|\.cxx$|\.C$', f)]
nvcc fatal   : Unsupported gpu architecture 'compute_100'
Target //xla/tools/multihost_hlo_runner:hlo_runner_main failed to build
Use --verbose_failures to see the command lines of failed build steps.

As @sergachev suggested, this problem can be solved by running the configuration with clang as:
``python /opt/xla/configure.py --backend cuda --cuda_compiler clang`

@Steboss Steboss requested a review from olupton January 22, 2025 11:07
@Steboss Steboss marked this pull request as draft January 22, 2025 14:17
@Steboss Steboss closed this Jan 24, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant