JuliaGPU / JuliaGPU/GPUArrays.jl

Kronecker product between `Diagonal` and `AbstractGPUArray`

Open
#585 8 comments 0 reactions 0 assignees View on GitHub
good first issue
Dominant language
Julia
Stars
450
Forks
104
Avg merge
1d 4h
Merged PRs (30d)
10

Description

The Kronecker product between `Diagonal` and `CUDA.CUSPARSE.AbstractCuSparseMatrix` works:

```julia
using LinearAlgebra
using CUDA
CUDA.allowscalar(false)

Ms = cu(sprand(ComplexF64, 4, 4, 0.5))
Id = I(2)
kron(Id, Ms)
```

Is it possible to also support Kronecker product between `Diagonal` and (dense) `AbstractGPUArray` ?
Because currently it's using scalar indexing:

```julia
Md = cu(rand(ComplexF64, 4, 4))
Id = I(2)
kron(Id, Md)
```

I post it here because this should also help for other GPU backends, `Metal` etc.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.