mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
68f3fec744
* python_test: fix cython_examples in doc/ and tests/ * update setup.py to parse the bazel version string better * all: centralize all python deps into stackable requirements files in python/ * format * Move cython test into the proper package * Add cross-reference dependency comments for requirements and setup.py * re-enable version pinning on CI, fix formatting * fix up torchvision version * fix case in shell
24 lines
819 B
Python
24 lines
819 B
Python
# flake8: noqa
|
|
import numpy
|
|
import pyximport
|
|
|
|
pyximport.install(setup_args={"include_dirs": numpy.get_include()})
|
|
|
|
from .cython_simple import simple_func, fib, fib_int, \
|
|
fib_cpdef, fib_cdef, simple_class
|
|
from .masked_log import masked_log
|
|
|
|
from .cython_blas import \
|
|
compute_self_corr_for_voxel_sel, \
|
|
compute_kernel_matrix, \
|
|
compute_single_self_corr_syrk, \
|
|
compute_single_self_corr_gemm, \
|
|
compute_corr_vectors, \
|
|
compute_single_matrix_multiplication
|
|
|
|
__all__ = [
|
|
"simple_func", "fib", "fib_int", "fib_cpdef", "fib_cdef", "simple_class",
|
|
"masked_log", "compute_self_corr_for_voxel_sel", "compute_kernel_matrix",
|
|
"compute_single_self_corr_syrk", "compute_single_self_corr_gemm",
|
|
"compute_corr_vectors", "compute_single_matrix_multiplication"
|
|
]
|