From b8b46d75e376f46b1f49da692daf5c23690d357f Mon Sep 17 00:00:00 2001 From: William Moses Date: Sun, 16 Jun 2024 17:17:15 -0400 Subject: [PATCH] Update kernel_launching.jl --- Manifest.toml | 59 +++++------------------------------ src/Utils/kernel_launching.jl | 2 +- 2 files changed, 9 insertions(+), 52 deletions(-) diff --git a/Manifest.toml b/Manifest.toml index 2c93e836cd..961740127e 100644 --- a/Manifest.toml +++ b/Manifest.toml @@ -1,8 +1,8 @@ # This file is machine-generated - editing it directly is not advised -julia_version = "1.10.2" +julia_version = "1.10.3" manifest_format = "2.0" -project_hash = "273a485e299a011d0a8b06266419fc09605b1e9a" +project_hash = "04d395caf937b0921325a77873167e8baa293a99" [[deps.AbstractFFTs]] deps = ["LinearAlgebra"] @@ -166,7 +166,7 @@ weakdeps = ["Dates", "LinearAlgebra"] [[deps.CompilerSupportLibraries_jll]] deps = ["Artifacts", "Libdl"] uuid = "e66e0078-7015-5450-92f7-15fbd957f2ae" -version = "1.1.0+0" +version = "1.1.1+0" [[deps.ConstructionBase]] deps = ["LinearAlgebra"] @@ -259,37 +259,6 @@ git-tree-sha1 = "71c79e77221ab3a29918aaf6db4f217b89138608" uuid = "b305315f-e792-5b7a-8f41-49f472929428" version = "1.0.1" -[[deps.Enzyme]] -deps = ["CEnum", "EnzymeCore", "Enzyme_jll", "GPUCompiler", "LLVM", "Libdl", "LinearAlgebra", "ObjectFile", "Preferences", "Printf", "Random"] -path = "../Enzyme.jl" -uuid = "7da242da-08ed-463a-9acd-ee780be4f1d9" -version = "0.12.13" - - [deps.Enzyme.extensions] - EnzymeChainRulesCoreExt = "ChainRulesCore" - EnzymeSpecialFunctionsExt = "SpecialFunctions" - EnzymeStaticArraysExt = "StaticArrays" - - [deps.Enzyme.weakdeps] - ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" - SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b" - StaticArrays = "90137ffa-7385-5640-81b9-e52037218182" - -[[deps.EnzymeCore]] -path = "../Enzyme.jl/lib/EnzymeCore" -uuid = "f151be2c-9106-41f4-ab19-57ee4f262869" -version = "0.7.4" -weakdeps = ["Adapt"] - - [deps.EnzymeCore.extensions] - AdaptExt = "Adapt" - -[[deps.Enzyme_jll]] -deps = ["Artifacts", "JLLWrappers", "LazyArtifacts", "Libdl", "TOML"] -git-tree-sha1 = "e3ece7b5fb991252abd138a2978e970063fc1412" -uuid = "7cc45869-7501-5eee-bdea-0790c847d4ef" -version = "0.0.121+0" - [[deps.ExprTools]] git-tree-sha1 = "27415f162e6028e81c72b82ef756bf321213b6ec" uuid = "e2ba6199-217a-4e67-a87a-7c52f15ade04" @@ -452,16 +421,16 @@ version = "0.2.1+0" [[deps.KernelAbstractions]] deps = ["Adapt", "Atomix", "InteractiveUtils", "LinearAlgebra", "MacroTools", "PrecompileTools", "Requires", "SparseArrays", "StaticArrays", "UUIDs", "UnsafeAtomics", "UnsafeAtomicsLLVM"] -git-tree-sha1 = "054e62cad04568b0fdd929636dc37d309a10379e" -repo-rev = "vc/varargs" -repo-url = "https://github.com/JuliaGPU/KernelAbstractions.jl.git" +git-tree-sha1 = "db02395e4c374030c53dc28f3c1d33dec35f7272" uuid = "63c18a36-062a-441e-b654-da1e3ab1ce7c" -version = "0.9.20" -weakdeps = ["EnzymeCore"] +version = "0.9.19" [deps.KernelAbstractions.extensions] EnzymeExt = "EnzymeCore" + [deps.KernelAbstractions.weakdeps] + EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869" + [[deps.LLVM]] deps = ["CEnum", "LLVMExtra_jll", "Libdl", "Preferences", "Printf", "Requires", "Unicode"] git-tree-sha1 = "065c36f95709dd4a676dc6839a35d6fa6f192f24" @@ -669,12 +638,6 @@ version = "3.7.2+0" uuid = "ca575930-c2e3-43a9-ace4-1e988b2c1908" version = "1.2.0" -[[deps.ObjectFile]] -deps = ["Reexport", "StructIO"] -git-tree-sha1 = "195e0a19842f678dd3473ceafbe9d82dfacc583c" -uuid = "d8793406-e978-5875-9003-1fc021f44a92" -version = "0.4.1" - [[deps.OffsetArrays]] git-tree-sha1 = "e64b4f5ea6b7389f6f046d13d4896a8f9c1ba71e" uuid = "6fe1bfb0-de20-5000-8ca7-80f57d26f881" @@ -996,12 +959,6 @@ weakdeps = ["Adapt", "GPUArraysCore", "SparseArrays", "StaticArrays"] StructArraysSparseArraysExt = "SparseArrays" StructArraysStaticArraysExt = "StaticArrays" -[[deps.StructIO]] -deps = ["Test"] -git-tree-sha1 = "010dc73c7146869c042b49adcdb6bf528c12e859" -uuid = "53d494c1-5632-5724-8f4c-31dff12d585f" -version = "0.3.0" - [[deps.StructTypes]] deps = ["Dates", "UUIDs"] git-tree-sha1 = "ca4bccb03acf9faaf4137a9abc1881ed1841aa70" diff --git a/src/Utils/kernel_launching.jl b/src/Utils/kernel_launching.jl index 7d8b5e8393..b0d3338669 100644 --- a/src/Utils/kernel_launching.jl +++ b/src/Utils/kernel_launching.jl @@ -113,7 +113,7 @@ end Launches `kernel!`, with arguments `args` and keyword arguments `kwargs`, over the `dims` of `grid` on the architecture `arch`. kernels run on the default stream """ -@inline function launch!(arch, grid, workspec, kernel!, kernel_args...; +function launch!(arch, grid, workspec, kernel!, kernel_args...; include_right_boundaries = false, reduced_dimensions = (), location = nothing,