add back allowscalar

sunxd3 · sunxd3 · commit 2fbc6faf88ea · 2025-03-14T06:10:36.000Z
diff --git a/test/ext/CUDA/cuda.jl b/test/ext/CUDA/cuda.jl
@@ -34,6 +34,7 @@ using Bijectors, CUDA, Distributions, Flux, LinearAlgebra, Test
     ]
 
     @testset "$dist" for dist in dists
+        CUDA.allowscalar(true)
         x = NormalizingFlows.rand_device(CUDA.default_rng(), dist)
         xs = NormalizingFlows.rand_device(CUDA.default_rng(), dist, 100)
         @test_nowarn logpdf(dist, x)

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@ using Bijectors, CUDA, Distributions, Flux, LinearAlgebra, Test`
`34`	`34`	`]`
`35`	`35`
`36`	`36`	`@testset "$dist" for dist in dists`
	`37`	`+ CUDA.allowscalar(true)`
`37`	`38`	`x = NormalizingFlows.rand_device(CUDA.default_rng(), dist)`
`38`	`39`	`xs = NormalizingFlows.rand_device(CUDA.default_rng(), dist, 100)`
`39`	`40`	`@test_nowarn logpdf(dist, x)`