Implement p-norm.

Author: maleadt

Project.toml

@@ -14,10 +14,3 @@ Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
 AbstractFFTs = "0.4, 0.5, 1.0"
 Adapt = "2.0, 3.0"
 julia = "1.5"
-
-[extras]
-FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
-
-[targets]
-test = ["Test", "FillArrays"]

src/host/linalg.jl

@@ -211,3 +211,16 @@ end
 # TODO: implementation without the memory copy
 LinearAlgebra.permutedims!(dest::AbstractGPUArray, src::AbstractGPUArray, perm) =
     permutedims!(dest, src, Tuple(perm))
+
+
+## norm
+
+function LinearAlgebra.norm(v::AbstractGPUArray{T}, p::Real=2) where {T}
+    if p == Inf
+        maximum(abs.(v))
+    elseif p == -Inf
+        minimum(abs.(v))
+    else
+        mapreduce(x->abs(x)^p, +, v; init=zero(T))^(1/p)
+    end
+end

test/Project.toml

@@ -0,0 +1,6 @@
+[deps]
+Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
+FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

test/testsuite/linalg.jl

@@ -1,5 +1,5 @@
 @testsuite "linear algebra" AT->begin
-    @testset "adjoint and transpose" begin
+    @testset "adjoint and trspose" begin
         @test compare(adjoint, AT, rand(Float32, 32, 32))
         @test compare(adjoint!, AT, rand(Float32, 32, 32), rand(Float32, 32, 32))
         @test compare(transpose, AT, rand(Float32, 32, 32))
@@ -114,4 +114,11 @@
         @test compare(rmul!, AT, rand(T, a), Ref(rand(T)))
         @test compare(lmul!, AT, Ref(rand(T)), rand(T, b))
     end
+
+    @testset "$p-norm($sz x $T)" for sz in [(2,), (2,2), (2,2,2)],
+                                     p in Any[1, 2, 3, Inf, -Inf],
+                                     T in supported_eltypes()
+        range = T <: Integer ? (T(1):T(10)) : T # prevent integer overflow
+        @test compare(norm, AT, rand(range, sz), Ref(p))
+    end
 end