Skip to content

3D and 4D support for Linear Algebra functions #483

Description

@pavanky

Do this inside a for loop with offsets. This will be needed for GFOR

CUDA

  • matmul
  • [ ] dot
  • lu
  • qr
  • cholesky
  • solve
  • inverse

OpenCL

  • matmul
  • [ ] dot
  • lu
  • qr
  • cholesky
  • solve
  • inverse

CPU

  • matmul
  • [ ] dot
  • lu
  • qr
  • cholesky
  • solve
  • inverse

The first draft can be done inside a for loop using multiple queues / streams.

Note: dot product function has been removed from the above list since a batch dot product can be easily done using matmul function.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions