File size: 835 Bytes
587b6c9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# [internal] load cc_fuzz_target.bzl
# [internal] load cc_proto_library.bzl
# [internal] load android_cc_test:def.bzl

load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")

package(default_visibility = [":__subpackages__"])

licenses(["notice"])

# To run all cc_tests in this directory:
# bazel test //:all

# [internal] Command to run dsp_util_android_test.

# [internal] Command to run lyra_integration_android_test.

exports_files(
    srcs = [
        "wavegru_mod.cc",
    ],
)

pybind_extension(
    name = "wavegru_mod",  # This name is not actually created!
    srcs = ["wavegru_mod.cc"],
    deps = [
        "//sparse_matmul",
    ],
)

py_library(
    name = "wavegru_mod",
    data = [":wavegru_mod.so"],
)

py_binary(
    name = "wavegru",
    srcs = ["wavegru.py"],
    deps = [
        ":wavegru_mod"
    ],
)