From 2efbebde42e96ea6253acec60d750b0d120cd19c Mon Sep 17 00:00:00 2001 From: mtfishman Date: Sun, 12 Oct 2025 00:25:50 -0400 Subject: [PATCH] [NDTensors] Define denseblocks on Dense --- NDTensors/Project.toml | 2 +- NDTensors/src/dense/densetensor.jl | 2 ++ NDTensors/test/test_dense.jl | 4 ++++ 3 files changed, 7 insertions(+), 1 deletion(-) diff --git a/NDTensors/Project.toml b/NDTensors/Project.toml index e5edea4365..138c276bfc 100644 --- a/NDTensors/Project.toml +++ b/NDTensors/Project.toml @@ -1,7 +1,7 @@ name = "NDTensors" uuid = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf" authors = ["Matthew Fishman "] -version = "0.4.14" +version = "0.4.15" [deps] Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697" diff --git a/NDTensors/src/dense/densetensor.jl b/NDTensors/src/dense/densetensor.jl index 17aabaecc7..5cb547b4c3 100644 --- a/NDTensors/src/dense/densetensor.jl +++ b/NDTensors/src/dense/densetensor.jl @@ -75,6 +75,8 @@ end # Useful for using Base Array functions array(T::DenseTensor) = convert(Array, T) +denseblocks(T::DenseTensor) = T + function diagview(T::DenseTensor) return diagview(array(T)) end diff --git a/NDTensors/test/test_dense.jl b/NDTensors/test/test_dense.jl index c17b816f35..4f62abda43 100644 --- a/NDTensors/test/test_dense.jl +++ b/NDTensors/test/test_dense.jl @@ -1,5 +1,6 @@ @eval module $(gensym()) using NDTensors +using NDTensors: denseblocks using NDTensors.MetalExtensions: mtl using Test: @testset, @test, @test_throws, @test_broken using GPUArraysCore: @allowscalar @@ -30,6 +31,9 @@ NDTensors.dim(i::MyInd) = i.dim @test dims(A[2:3, 2:end]) == (2, 3) @test dims(A[3, 2:end]) == (3,) + @test dense(A) ≡ A + @test denseblocks(A) ≡ A + randn!(A) @test ndims(A) == 2