|
41 | 41 |
|
42 | 42 | const REACTANT_TEST_GROUP = lowercase(get(ENV, "REACTANT_TEST_GROUP", "all"))
|
43 | 43 |
|
44 |
| -# @testset ExtendedTestSet "Reactant.jl Tests" begin |
45 |
| -# if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "core" |
46 |
| -# # @safetestset "Layout" include("layout.jl") |
47 |
| -# # @safetestset "Tracing" include("tracing.jl") |
48 |
| -# @safetestset "Basic" include("basic.jl") |
49 |
| -# # @safetestset "Autodiff" include("autodiff.jl") |
50 |
| -# # @safetestset "Complex" include("complex.jl") |
51 |
| -# # @safetestset "Broadcast" include("bcast.jl") |
52 |
| -# # @safetestset "Struct" include("struct.jl") |
53 |
| -# # @safetestset "Closure" include("closure.jl") |
54 |
| -# # @safetestset "Compile" include("compile.jl") |
55 |
| -# # @safetestset "Buffer Donation" include("buffer_donation.jl") |
56 |
| -# # @safetestset "Shortcuts to MLIR ops" include("ops.jl") |
57 |
| -# # @safetestset "Wrapped Arrays" include("wrapped_arrays.jl") |
58 |
| -# # @safetestset "Control Flow" include("control_flow.jl") |
59 |
| -# # @safetestset "Sorting" include("sorting.jl") |
60 |
| -# # @safetestset "Indexing" include("indexing.jl") |
61 |
| -# # if !Sys.isapple() |
62 |
| -# # @safetestset "Custom Number Types" include("custom_number_types.jl") |
63 |
| -# # end |
64 |
| -# # @safetestset "Sharding" include("sharding.jl") |
| 44 | +@testset ExtendedTestSet "Reactant.jl Tests" begin |
| 45 | + if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "core" |
| 46 | + # @safetestset "Layout" include("layout.jl") |
| 47 | + # @safetestset "Tracing" include("tracing.jl") |
| 48 | + @safetestset "Basic" include("basic.jl") |
| 49 | + # @safetestset "Autodiff" include("autodiff.jl") |
| 50 | + # @safetestset "Complex" include("complex.jl") |
| 51 | + # @safetestset "Broadcast" include("bcast.jl") |
| 52 | + # @safetestset "Struct" include("struct.jl") |
| 53 | + # @safetestset "Closure" include("closure.jl") |
| 54 | + # @safetestset "Compile" include("compile.jl") |
| 55 | + # @safetestset "Buffer Donation" include("buffer_donation.jl") |
| 56 | + # @safetestset "Shortcuts to MLIR ops" include("ops.jl") |
| 57 | + # @safetestset "Wrapped Arrays" include("wrapped_arrays.jl") |
| 58 | + # @safetestset "Control Flow" include("control_flow.jl") |
| 59 | + # @safetestset "Sorting" include("sorting.jl") |
| 60 | + # @safetestset "Indexing" include("indexing.jl") |
| 61 | + # if !Sys.isapple() |
| 62 | + # @safetestset "Custom Number Types" include("custom_number_types.jl") |
| 63 | + # end |
| 64 | + # @safetestset "Sharding" include("sharding.jl") |
65 | 65 |
|
66 |
| -# # @testset "IFRT" begin |
67 |
| -# # @safetestset "IFRT Low-Level API" include("ifrt/low_level.jl") |
68 |
| -# # end |
69 |
| -# end |
| 66 | + # @testset "IFRT" begin |
| 67 | + # @safetestset "IFRT Low-Level API" include("ifrt/low_level.jl") |
| 68 | + # end |
| 69 | + end |
70 | 70 |
|
71 |
| -# if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "integration" |
72 |
| -# # @safetestset "CUDA" include("integration/cuda.jl") |
73 |
| -# # @safetestset "KernelAbstractions" include("integration/kernelabstractions.jl") |
74 |
| -# # @safetestset "Linear Algebra" include("integration/linear_algebra.jl") |
75 |
| -# # @safetestset "OffsetArrays" include("integration/offsetarrays.jl") |
76 |
| -# # @safetestset "AbstractFFTs" include("integration/fft.jl") |
77 |
| -# @safetestset "SpecialFunctions" include("integration/special_functions.jl") |
78 |
| -# # @safetestset "Random" include("integration/random.jl") |
79 |
| -# # @safetestset "Python" include("integration/python.jl") |
80 |
| -# end |
| 71 | + if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "integration" |
| 72 | + # @safetestset "CUDA" include("integration/cuda.jl") |
| 73 | + # @safetestset "KernelAbstractions" include("integration/kernelabstractions.jl") |
| 74 | + # @safetestset "Linear Algebra" include("integration/linear_algebra.jl") |
| 75 | + # @safetestset "OffsetArrays" include("integration/offsetarrays.jl") |
| 76 | + # @safetestset "AbstractFFTs" include("integration/fft.jl") |
| 77 | + @safetestset "SpecialFunctions" include("integration/special_functions.jl") |
| 78 | + # @safetestset "Random" include("integration/random.jl") |
| 79 | + # @safetestset "Python" include("integration/python.jl") |
| 80 | + end |
81 | 81 |
|
82 |
| -# if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "neural_networks" |
83 |
| -# @testset "Neural Networks" begin |
84 |
| -# @safetestset "NNlib Primitives" include("nn/nnlib.jl") |
85 |
| -# @safetestset "Flux.jl Integration" include("nn/flux.jl") |
86 |
| -# if Sys.islinux() |
87 |
| -# @safetestset "LuxLib Primitives" include("nn/luxlib.jl") |
88 |
| -# @safetestset "Lux Integration" include("nn/lux.jl") |
89 |
| -# end |
90 |
| -# end |
91 |
| -# end |
92 |
| -# end |
| 82 | + if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "neural_networks" |
| 83 | + @testset "Neural Networks" begin |
| 84 | + @safetestset "NNlib Primitives" include("nn/nnlib.jl") |
| 85 | + @safetestset "Flux.jl Integration" include("nn/flux.jl") |
| 86 | + if Sys.islinux() |
| 87 | + @safetestset "LuxLib Primitives" include("nn/luxlib.jl") |
| 88 | + @safetestset "Lux Integration" include("nn/lux.jl") |
| 89 | + end |
| 90 | + end |
| 91 | + end |
| 92 | +end |
0 commit comments