conda-forge / jaxlib-feedstock

A conda-smithy repository for jaxlib.
BSD 3-Clause "New" or "Revised" License
16 stars 24 forks source link

jaxlib 0.4.27 #268

Closed xhochy closed 3 months ago

xhochy commented 3 months ago

Fixes #267

Checklist

conda-forge-webservices[bot] commented 3 months ago

Hi! This is the friendly automated conda-forge-linting service.

I just wanted to let you know that I linted all conda-recipes in your PR (recipe) and found it was in an excellent condition.

xhochy commented 3 months ago

aarch64 builds fail with the same error as the 0.4.26 build:

2024-07-02T11:12:14.5487144Z ImportError: /home/conda/feedstock_root/build_artifacts/jaxlib_1719908436950/_test_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold/lib/python3.10/site-packages/jaxlib/xla_extension.so: undefined symbol: LLVMInitializeX86Target
h-vetinari commented 3 months ago

aarch64 builds fail with [...] undefined symbol: LLVMInitializeX86Target

Very weird that anything on aarch64 would require something from x86 in the first place.

xhochy commented 3 months ago

Current error:

ERROR: /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_build_env/share/bazel/07d1bb0506f141129674d0fd20e199ef/external/xla/xla/stream_executor/gpu/BUILD:334:19: Compiling xla/stream_executor/gpu/gpu_timer_kernel_cuda.cu.cc failed: (Exit 1): crosstool_wrapper_driver_is_not_gcc failed: error executing command (from target @xla//xla/stream_executor/gpu:gpu_timer_kernel_cuda)                   18:28:40 [57/1624]
  (cd /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_build_env/share/bazel/07d1bb0506f141129674d0fd20e199ef/execroot/__main__ && \
  exec env - \
    CUDA_TOOLKIT_PATH=/usr/local/cuda,/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac \
    CUDNN_INSTALL_PATH=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac \
    GCC_HOST_COMPILER_PATH=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_build_env/bin/x86_64-conda-linux-gnu-gcc \
    GCC_HOST_COMPILER_PREFIX=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_build_env/bin \
    PATH=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_build_env/bin:/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/bin:/opt/conda/condabin:/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208
747/_build_env/bin:/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/bin:/opt/conda/bin:/opt/conda/condabin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/home/conda/bin:/usr/local/cuda/bin \                    PWD=/proc/self/cwd \
    TF_CUDA_COMPUTE_CAPABILITIES=sm_35,sm_50,sm_60,sm_62,sm_70,sm_72,sm_75,sm_80,sm_86,sm_87,sm_89,sm_90,compute_90 \
    TF_CUDA_PATHS=/usr/local/cuda,/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac,/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placeho
ld_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac \
    TF_CUDA_VERSION=11.8 \
    TF_CUDNN_VERSION=8 \
  bazel_toolchain/crosstool_wrapper_driver_is_not_gcc -MD -MF bazel-out/k8-opt/bin/external/xla/xla/stream_executor/gpu/_objs/gpu_timer_kernel_cuda/gpu_timer_kernel_cuda.cu.pic.d '-frandom-seed=bazel-out/k8-opt/bin/external/xla/xla/stream_executor/gpu/_objs/gpu_timer_kernel_cuda/gpu_timer_kernel_cuda.cu.pic.o' -fPIC '-DEIGEN_MAX_ALIGN_BYTES=64' -DEIGEN_ALLOW_UNALIGNED_SCALARS '-DEIGEN_USE_AVX512_GEMM_KERNELS=0' -DHAVE_SYS_UI
O_H -DTF_USE_SNAPPY '-DGOOGLE_CUDA=1' '-DBAZEL_CURRENT_REPOSITORY="xla"' -iquote external/xla -iquote bazel-out/k8-opt/bin/external/xla -iquote external/tsl -iquote bazel-out/k8-opt/bin/external/tsl -iquote external/eigen_archive -iquote bazel-out/k8-opt/bin/external/eigen_archive -iquote external/ml_dtypes -iquote bazel-out/k8-opt/bin/external/ml_dtypes -iquote external/com_google_absl -iquote bazel-out/k8-opt/bin/external/
com_google_absl -iquote external/nsync -iquote bazel-out/k8-opt/bin/external/nsync -iquote external/local_config_cuda -iquote bazel-out/k8-opt/bin/external/local_config_cuda -iquote external/com_google_protobuf -iquote bazel-out/k8-opt/bin/external/com_google_protobuf -iquote external/zlib -iquote bazel-out/k8-opt/bin/external/zlib -iquote external/double_conversion -iquote bazel-out/k8-opt/bin/external/double_conversion -iq
uote external/snappy -iquote bazel-out/k8-opt/bin/external/snappy -iquote external/com_googlesource_code_re2 -iquote bazel-out/k8-opt/bin/external/com_googlesource_code_re2 -Ibazel-out/k8-opt/bin/external/ml_dtypes/_virtual_includes/float8 -Ibazel-out/k8-opt/bin/external/ml_dtypes/_virtual_includes/intn -Ibazel-out/k8-opt/bin/external/local_config_cuda/cuda/_virtual_includes/cuda_headers_virtual -isystem external/eigen_archi
ve -isystem bazel-out/k8-opt/bin/external/eigen_archive -isystem external/eigen_archive/mkl_include -isystem bazel-out/k8-opt/bin/external/eigen_archive/mkl_include -isystem external/ml_dtypes -isystem bazel-out/k8-opt/bin/external/ml_dtypes -isystem external/ml_dtypes/ml_dtypes -isystem bazel-out/k8-opt/bin/external/ml_dtypes/ml_dtypes -isystem external/nsync/public -isystem bazel-out/k8-opt/bin/external/nsync/public -isyst
em external/local_config_cuda/cuda -isystem bazel-out/k8-opt/bin/external/local_config_cuda/cuda -isystem external/local_config_cuda/cuda/cuda/include -isystem bazel-out/k8-opt/bin/external/local_config_cuda/cuda/cuda/include -isystem external/com_google_protobuf/src -isystem bazel-out/k8-opt/bin/external/com_google_protobuf/src -isystem /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placeho
ld_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include '-march=nocona' '-mtune=haswell' -ftree-vectorize -fPIC -fstack-protector-strong -fno-plt -O2 -ffunction-sections -pipe -isystem /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_pl
acehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include '-fdebug-prefix-map=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/work=/usr/local/src/conda/jaxlib-0.4.27' '-fdebug-prefix-map=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_p
lacehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac=/usr/local/src/conda-prefix' -isystem /usr/local/cuda/include -DNDEBUG -DNDEBUG '-D_FORTIFY_SOURCE=2' -O2 -isystem /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehol
d_placehold_placehold_placehold_placehold_placehold_plac/include -isystem /usr/local/cuda/include -fvisibility-inlines-hidden '-fmessage-length=0' '-march=nocona' '-mtune=haswell' -ftree-vectorize -fPIC -fstack-protector-strong -fno-plt -O2 -ffunction-sections -pipe -isystem /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_p
lacehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include '-fdebug-prefix-map=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/work=/usr/local/src/conda/jaxlib-0.4.27' '-fdebug-prefix-map=/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_
placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac=/usr/local/src/conda-prefix' -isystem /usr/local/cuda/include -DNDEBUG -DNDEBUG '-D_FORTIFY_SOURCE=2' -O2 -isystem /home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placeho
ld_placehold_placehold_plac/include -isystem /usr/local/cuda/include '-fvisibility=hidden' -Wno-sign-compare -Wno-unknown-warning-option -Wno-stringop-truncation -Wno-array-parameter '-DMLIR_PYTHON_PACKAGE_PREFIX=jaxlib.mlir.' '-std=c++17' -x cuda '-DGOOGLE_CUDA=1' '--cuda-gpu-arch=sm_35' '--cuda-gpu-arch=sm_50' '--cuda-gpu-arch=sm_60' '--cuda-gpu-arch=sm_62' '--cuda-gpu-arch=sm_70' '--cuda-gpu-arch=sm_72' '--cuda-gpu-arch=s
m_75' '--cuda-gpu-arch=sm_80' '--cuda-gpu-arch=sm_86' '--cuda-gpu-arch=sm_87' '--cuda-gpu-arch=sm_89' '--cuda-gpu-arch=sm_90' '--cuda-include-ptx=sm_90' '--cuda-gpu-arch=sm_90' '-Xcuda-fatbinary=--compress-all' '-nvcc_options=expt-relaxed-constexpr' -c external/xla/xla/stream_executor/gpu/gpu_timer_kernel_cuda.cu.cc -o bazel-out/k8-opt/bin/external/xla/xla/stream_executor/gpu/_objs/gpu_timer_kernel_cuda/gpu_timer_kernel_cuda
.cu.pic.o)
# Configuration: dd29da5933a58ab83580e7b47661e54c6634e54f1deafc85c0860a2608211325
# Execution platform: @local_execution_config_platform//:platform
nvcc warning : The 'compute_35', 'compute_37', 'sm_35', and 'sm_37' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).
nvcc warning : The 'compute_35', 'compute_37', 'sm_35', and 'sm_37' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h: In constructor 'absl::lts_20240116::str_format_internal::FormatSpecTemplate<Args>::FormatSpecTemplate(const absl::lts_20240
116::str_format_internal::ExtendedParsedFormat<absl::lts_20240116::FormatConversionCharSet(C)...>&)':
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h:175:1: error: parse error in template argument list
  175 |     CheckArity<sizeof...(C), sizeof...(Args)>();
      | ^   ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h:175:63: error: expected ';' before ')' token
  175 |     CheckArity<sizeof...(C), sizeof...(Args)>();
      |                                                               ^
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h:176:138: error: template argument 1 is invalid
  176 |     CheckMatches<C...>(absl::make_index_sequence<sizeof...(C)>{});
      |                                                                                                                                          ^
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h:176:142: error: expected primary-expression before '{' token
  176 |     CheckMatches<C...>(absl::make_index_sequence<sizeof...(C)>{});
      |                                                                                                                                              ^
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h:176:142: error: expected ';' before '{' token
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/bind.h:176:144: error: expected primary-expression before ')' token
  176 |     CheckMatches<C...>(absl::make_index_sequence<sizeof...(C)>{});
      |                                                                                                                                                ^
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h: In instantiation of 'constexpr absl::lts_20240116::FormatConversionCharSet absl::lts_20240116::str_format_internal::Argument
ToConv() [with Arg = unsigned int]':
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/str_format.h:279:156:   required by substitution of 'template<class ... Args> using FormatSpec = absl::lts_20240116::str_format_internal::FormatSpecTem
plate<absl::lts_20240116::FormatConversionCharSet((ArgumentToConv<Args>)())...> [with Args = {unsigned int}]'
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/str_format.h:399:1:   required by substitution of 'template<class ... Args> absl::lts_20240116::str_format_internal::Streamable absl::lts_20240116::Str
eamFormat(absl::lts_20240116::FormatSpec<Args ...>&, const Args& ...) [with Args = {unsigned int}]'
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/crc/crc32c.h:184:34:   required from here
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:435:43: error: no matching function for call to 'ExtractCharSet(ConvResult)'
  435 |   return absl::str_format_internal::ExtractCharSet(ConvResult{});
      |        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:206:1: note: candidate: 'template<absl::lts_20240116::FormatConversionCharSet C> constexpr absl::lts_20240116::FormatConversi
onCharSet absl::lts_20240116::str_format_internal::ExtractCharSet(absl::lts_20240116::FormatConvertResult<(absl::lts_20240116::FormatConversionCharSet)(C)>)'
  206 | constexpr FormatConversionCharSet ExtractCharSet(FormatConvertResult<C>) {
      | ^~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:206:1: note:   template argument deduction/substitution failed:
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:435:43: note:   couldn't deduce template parameter 'C'
  435 |   return absl::str_format_internal::ExtractCharSet(ConvResult{});
      |        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:211:1: note: candidate: 'template<absl::lts_20240116::FormatConversionCharSet C> constexpr absl::lts_20240116::FormatConversi
onCharSet absl::lts_20240116::str_format_internal::ExtractCharSet(absl::lts_20240116::str_format_internal::ArgConvertResult<(absl::lts_20240116::FormatConversionCharSet)(C)>)'
  211 | constexpr FormatConversionCharSet ExtractCharSet(ArgConvertResult<C>) {
      | ^~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:211:1: note:   template argument deduction/substitution failed:
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/internal/str_format/arg.h:435:43: note:   couldn't deduce template parameter 'C'
  435 |   return absl::str_format_internal::ExtractCharSet(ConvResult{});
      |        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/crc/crc32c.h: In function 'std::ostream& absl::lts_20240116::operator<<(std::ostream&, absl::lts_20240116::crc32c_t)':
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/crc/crc32c.h:184:34: error: no matching function for call to 'StreamFormat(const char [5], uint32_t)'
  184 |   return os << absl::StreamFormat("%08x", static_cast<uint32_t>(crc));
      |                ~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/str_format.h:399:1: note: candidate: 'template<class ... Args> absl::lts_20240116::str_format_internal::Streamable absl::lts_20240116::StreamFormat(abs
l::lts_20240116::FormatSpec<Args ...>&, const Args& ...)'
  399 | ABSL_MUST_USE_RESULT str_format_internal::Streamable StreamFormat(
      | ^~~~~~~~~~~~
/home/conda/feedstock_root/build_artifacts/jaxlib_1720117208747/_h_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_plac/include/absl/strings/str_format.h:399:1: note:   substitution of deduced template arguments resulted in errors seen above
Target //jaxlib/tools:build_wheel failed to build
xhochy commented 3 months ago

@h-vetinari @conda-forge/jaxlib @conda-forge/jax I will drop CUDA 11.8 for now here. I cannot get it to build locally. I commented everything in xla/stream_executor/gpu/gpu_timer_kernel_cuda.cu.cc except a single line with #include "absl/status/status.h" and still get the above error. Omitting this include from any CUDA kernel is not an option. We can retry it in newer jaxlib releases. But for now, I don't see a viable path where we could have new jaxlib builds and CUDA 11.8 at the same time.