mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update ducc to commit: 2b2cead005e08d2632478e831d7f45da754162dc

Browse Source 
NOTE: this version of DUCC has a breaking change, where the fft.h header
no longer contains the definitions of many fft functions - instead they exist
within fft1d_impl.h and fftnd_impl.h.
PiperOrigin-RevId: 554567641
This commit is contained in:
Antonio Sanchez 2023-08-07 13:06:04 -07:00 committed by jax authors
parent 079ecfbf20
commit a600020346
3 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/cpu/ducc_fft_kernels.cc
View File

@ -16,6 +16,8 @@ limitations under the License.
#include <complex>
#include "ducc/src/ducc0/fft/fft.h"
#include "ducc/src/ducc0/fft/fft1d_impl.h" // NOLINT: required for fft definitions.
#include "ducc/src/ducc0/fft/fftnd_impl.h" // NOLINT: required for fft definitions.
#include "flatbuffers/flatbuffers.h"
#include "jaxlib/cpu/ducc_fft_generated.h"
#include "xla/service/custom_call_status.h"
@ -137,4 +139,4 @@ void DynamicDuccFft(void *out, void **in, XlaCustomCallStatus *) {
descriptor->forward(), *dynamic_scale);
}
} // namespace jax
} // namespace jax

15
third_party/ducc/BUILD.bazel vendored
View File

@ -2,29 +2,32 @@ 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/fft/fft1d.h",
"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/threading.h",
"src/ducc0/infra/useful_macros.h",
"src/ducc0/math/cmplx.h",
"src/ducc0/math/unity_roots.h",
],
hdrs = ["src/ducc0/fft/fft.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",
],
includes = ["src"],
)

12
third_party/ducc/workspace.bzl vendored
View File

@ -17,13 +17,17 @@
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-356d619a4b5f6f8940d15913c14a043355ef23be",
sha256 = "d23eb2d06f03604867ad40af4fe92dec7cccc2c59f5119e9f01b35b973885c61",
strip_prefix = "ducc-{commit}".format(commit = DUCC_COMMIT),
sha256 = DUCC_SHA256,
urls = [
"https://github.com/mreineck/ducc/archive/356d619a4b5f6f8940d15913c14a043355ef23be.tar.gz",
"https://storage.googleapis.com/jax-releases/mirror/ducc/ducc-356d619a4b5f6f8940d15913c14a043355ef23be.tar.gz",
"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",
)