Update deps & minor changes

JuliaNeuralGraphics · Dec 14, 2023 · c47dde0 · c47dde0
1 parent 6553fb8
commit c47dde0
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/Project.toml b/Project.toml
@@ -27,7 +27,7 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
-AMDGPU = "0.6, 0.7"
+AMDGPU = "0.8"
 Adapt = "3.6"
 BSON = "0.3"
 BenchmarkTools = "1.3.2"
@@ -40,7 +40,7 @@ ImageTransformations = "0.10"
 JSON = "0.21"
 JpegTurbo = "0.1"
 KernelAbstractions = "0.9"
-NerfUtils = "0.1.2"
+NerfUtils = "0.1.4"
 Preferences = "1.4"
 Quaternions = "0.7"
 Rotations = "1.6"

diff --git a/src/marching_cubes/marching_cubes.jl b/src/marching_cubes/marching_cubes.jl
@@ -165,8 +165,8 @@ function marching_cubes(
         "but is instead $(size(log_densities))")
 
     Backend = get_backend(log_densities)
-    counters = KernelAbstractions.zeros(Backend, UInt32, 2)
-    generation_counters = KernelAbstractions.zeros(Backend, UInt32, 2)
+    counters = adapt(Backend, zeros(UInt32, 2))
+    generation_counters = adapt(Backend, zeros(UInt32, 2))
     vertex_grid = allocate(Backend, Int32, (3, ndrange...))
     fill!(vertex_grid, Int32(-1))
 

diff --git a/src/renderer/utils.jl b/src/renderer/utils.jl
@@ -139,8 +139,8 @@ function materialize!(
     kab = get_backend(bundle)
     n_alive = Int(Array(bundle.alive_counter)[1])
 
-    steps_counter = KernelAbstractions.zeros(kab, UInt32, 1)
-    rays_counter = KernelAbstractions.zeros(kab, UInt32, 1)
+    steps_counter = adapt(kab, zeros(UInt32, 1))
+    rays_counter = adapt(kab, zeros(UInt32, 1))
 
     (; alive_rays) = alive(bundle)
     n_levels::UInt32 = get_n_levels(occupancy)

diff --git a/src/sampler.jl b/src/sampler.jl
@@ -97,8 +97,8 @@ function materialize!(
     T <: AbstractVector{SVector{3, Float32}},
 }
     kab = get_backend(occupancy)
-    steps_counter = KernelAbstractions.zeros(kab, UInt32, (1,))
-    rays_counter = KernelAbstractions.zeros(kab, UInt32, (1,))
+    steps_counter = adapt(kab, zeros(UInt32, 1))
+    rays_counter = adapt(kab, zeros(UInt32, 1))
 
     resolution::UInt32 = get_resolution(occupancy)
     n_levels::UInt32 = get_n_levels(occupancy)