From 259549da0ae6266ec2c8b77d52d9452c0eea4acf Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Sun, 18 Aug 2024 08:30:49 -0700 Subject: [PATCH] perf: add NNlib to benchmarks deps --- Project.toml | 2 +- benchmarks/Project.toml | 1 + benchmarks/setup.jl | 1 + test/shared_testsetup.jl | 2 +- 4 files changed, 4 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 6db0ebdb..0513ef2b 100644 --- a/Project.toml +++ b/Project.toml @@ -75,7 +75,7 @@ NNlib = "0.9.21" Octavian = "0.3.28" Polyester = "0.7.15" Random = "1.10" -Reexport = "1.2" +Reexport = "1" ReverseDiff = "1.15" SLEEFPirates = "0.6.43" Static = "0.8.4, 1" diff --git a/benchmarks/Project.toml b/benchmarks/Project.toml index e6436756..7fe762e6 100644 --- a/benchmarks/Project.toml +++ b/benchmarks/Project.toml @@ -3,6 +3,7 @@ BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf" InteractiveUtils = "b77e0a4c-d291-57a0-90e8-8db25a27a240" LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11" MLDataDevices = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40" +NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd" Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f" Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c" StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3" diff --git a/benchmarks/setup.jl b/benchmarks/setup.jl index f80ccf4b..06211e9d 100644 --- a/benchmarks/setup.jl +++ b/benchmarks/setup.jl @@ -1,4 +1,5 @@ using MLDataDevices, StableRNGs, Random +using NNlib using Zygote synchronize(::CPUDevice) = nothing diff --git a/test/shared_testsetup.jl b/test/shared_testsetup.jl index 6088d444..4cf27cfb 100644 --- a/test/shared_testsetup.jl +++ b/test/shared_testsetup.jl @@ -2,7 +2,7 @@ import Reexport: @reexport using LuxLib, MLDataDevices -@reexport using LuxTestUtils, StableRNGs, Test, Enzyme, Zygote +@reexport using LuxTestUtils, StableRNGs, Test, Enzyme, Zygote, NNlib LuxTestUtils.jet_target_modules!(["LuxLib"])