Basic tests for CuTensor type

Author: kshyatt

test/tensor.jl

@@ -11,6 +11,26 @@ else
 
 @test has_cutensor()
 
+@testset "CuTensor type basics" begin
+    N = 2
+    dmax = 2^div(18,N)
+    dims = rand(2:dmax, N)
+    p = randperm(N)
+    indsA = collect(('a':'z')[1:N])
+    dimsA = dims
+    A = rand(Float64, dimsA...)
+    dA = CuArray(A)
+    p = randperm(N)
+    indsA = collect(('a':'z')[1:N])
+    ctA = CuTensor(dA, indsA)
+    @test length(ctA) == length(A)
+    @test size(ctA) == size(A)
+    @test size(ctA, 1) == size(A, 1)
+    @test ndims(ctA) == ndims(A)
+    @test strides(ctA) == strides(A)
+    @test eltype(ctA) == eltype(A)
+end
+
 @testset "Elementwise binary" begin
     eltypes = ((Float16, Float16),
                #(Float16, Float32),