Merge pull request #274 from aterenin/master

Add inv for Triangular types and test.

src/host/linalg.jl

@@ -200,3 +200,12 @@ function LinearAlgebra.permutedims!(dest::AbstractGPUArray, src::AbstractGPUArra
     end
     return dest
 end
+
+
+## inv for Triangular
+for TR in (UpperTriangular, LowerTriangular, UnitUpperTriangular, UnitLowerTriangular)
+    @eval function Base.inv(x::$TR{<:Any,<:AbstractGPUArray})
+      out = typeof(parent(x))(I(size(x,1)))
+      $TR(LinearAlgebra.ldiv!(x,out))
+    end
+end

src/reference.jl

@@ -306,4 +306,14 @@ function GPUArrays.mapreducedim!(f, op, R::JLArray, A::Union{AbstractArray,Broad
     @allowscalar Base.reducedim!(op, R.data, map(f, A))
 end
 
+
+## LinearAlgebra
+
+using LinearAlgebra
+
+for TR in (UpperTriangular, LowerTriangular, UnitUpperTriangular, UnitLowerTriangular)
+    @eval LinearAlgebra.ldiv!(x::$TR{T,<:JLArray{T,2}}, y::JLArray{T,2}) where T<:Union{Complex{Float32}, Complex{Float64}, Float32, Float64} = JLArray(LinearAlgebra.ldiv!($TR(parent(x).data),y.data))
+    @eval LinearAlgebra.rdiv!(x::JLArray{T,2}, y::$TR{T,<:JLArray{T,2}}) where T<:Union{Complex{Float32}, Complex{Float64}, Float32, Float64} = JLArray(LinearAlgebra.rdiv!(x.data,$TR(parent(y).data)))
 end
+
+end

test/testsuite/linalg.jl

@@ -46,6 +46,10 @@ function test_linalg(AT)
             end
         end
 
+        @testset "inv for triangular" for TR in (UpperTriangular, LowerTriangular, UnitUpperTriangular, UnitLowerTriangular)
+            @test compare(x -> inv(TR(x)), AT, rand(Float32, 32, 32))
+        end
+
         @testset "permutedims" begin
             @test compare(x -> permutedims(x, (2, 1)), AT, rand(Float32, 2, 3))
             @test compare(x -> permutedims(x, (2, 1, 3)), AT, rand(Float32, 4, 5, 6))