1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00

Use TSL's import of DUCC.

This is a necessary first step before adding DUCC support to XLA,
otherwise the JAX tests in the XLA repo pull from JAX's copy,
which has slightly different build rules.

PiperOrigin-RevId: 576880208
This commit is contained in:
Antonio Sanchez 2023-10-26 08:25:33 -07:00 committed by jax authors
parent cd177fd566
commit 873cffc776
4 changed files with 1 additions and 70 deletions
WORKSPACE
jaxlib/cpu
third_party/ducc

@ -2,9 +2,6 @@
load("//third_party/xla:workspace.bzl", jax_xla_workspace = "repo")
jax_xla_workspace()
load("//third_party/ducc:workspace.bzl", ducc = "repo")
ducc()
load("@xla//:workspace4.bzl", "xla_workspace4")
xla_workspace4()

@ -79,7 +79,7 @@ cc_library(
":ducc_fft_flatbuffers_cc",
"@xla//xla/service:custom_call_status",
"@com_github_google_flatbuffers//:flatbuffers",
"@ducc",
"@ducc//:fft",
],
)

@ -1,33 +0,0 @@
licenses(["notice"])
package(default_visibility = ["//visibility:public"])
# The following files are dual-licensed as BSD 3 clause and GPLv2. We choose BSD 3 clause.
cc_library(
name = "ducc",
srcs = [
"src/ducc0/infra/aligned_array.h",
"src/ducc0/infra/error_handling.h",
"src/ducc0/infra/mav.h",
"src/ducc0/infra/misc_utils.h",
"src/ducc0/infra/simd.h",
"src/ducc0/infra/threading.cc",
"src/ducc0/infra/useful_macros.h",
"src/ducc0/math/cmplx.h",
"src/ducc0/math/unity_roots.h",
],
hdrs = [
"src/ducc0/fft/fft.h",
"src/ducc0/fft/fft1d_impl.h",
"src/ducc0/fft/fftnd_impl.h",
"src/ducc0/infra/threading.h",
],
copts = [
"-frtti",
"-fexceptions",
"-ffast-math",
],
features = ["-use_header_modules"],
include_prefix = "ducc",
includes = ["src"],
)

@ -1,33 +0,0 @@
# Copyright 2020 The JAX Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Bazel workspace for DUCC (CPU FFTs)."""
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
def repo():
# Attention: tools parse and update these lines.
DUCC_COMMIT = "2b2cead005e08d2632478e831d7f45da754162dc"
DUCC_SHA256 = "60719aa71d637dba594a03fed682bb6943dfffaa5557f8e8bb51228a295bbd79"
http_archive(
name = "ducc",
strip_prefix = "ducc-{commit}".format(commit = DUCC_COMMIT),
sha256 = DUCC_SHA256,
urls = [
"https://gitlab.mpcdf.mpg.de/mtr/ducc/-/archive/{commit}/ducc-{commit}.tar.gz".format(commit = DUCC_COMMIT),
"https://storage.googleapis.com/mirror.tensorflow.org/gitlab.mpcdf.mpg.de/mtr/ducc/-/archive/{commit}/ducc-{commit}.tar.gz".format(commit = DUCC_COMMIT),
],
build_file = "@//third_party/ducc:BUILD.bazel",
)