From 6976693d49414e0b0d2e23374cb27ab10ce951c8 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Mon, 28 Oct 2024 11:19:06 -0400 Subject: [PATCH] fix: missing import; fixes #179 (#180) --- Project.toml | 2 +- src/impl/Impl.jl | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 6f6005b7..a053be07 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "LuxLib" uuid = "82251201-b29d-42c6-8e01-566dec8acb11" authors = ["Avik Pal and contributors"] -version = "1.3.5" +version = "1.3.6" [deps] ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9" diff --git a/src/impl/Impl.jl b/src/impl/Impl.jl index b6a6a0d9..3bd59797 100644 --- a/src/impl/Impl.jl +++ b/src/impl/Impl.jl @@ -29,7 +29,8 @@ using ..Utils: Utils, NotaNumber, batchview, concrete_bias_act_output_eltype, co copy_drop_gradients, eltype_mismatch, expand_batchdim, maybe_reduce_BLAS_threads, ofeltype_array, only_derivative, remove_tracking, reset_BLAS_threads, run_ka_kernel, safe_eltype, safe_vec, safe_warning, - unsafe_known, unrolled_mapreduce, can_loopvec_args, @enzyme_alternative + unsafe_known, unrolled_mapreduce, can_loopvec_args, is_extension_loaded, + @enzyme_alternative using ..Traits: activation_intermediate_not_needed, activation_has_rrule, is_mutable_array, fuse_cpu_activation using ..System: explicit_blas_loaded, use_octavian, fits_in_l1cache, fits_in_l2cache,