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:
parent
cd177fd566
commit
873cffc776
@ -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",
|
||||
],
|
||||
)
|
||||
|
||||
|
33
third_party/ducc/BUILD.bazel
vendored
33
third_party/ducc/BUILD.bazel
vendored
@ -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"],
|
||||
)
|
33
third_party/ducc/workspace.bzl
vendored
33
third_party/ducc/workspace.bzl
vendored
@ -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",
|
||||
)
|
Loading…
x
Reference in New Issue
Block a user