Skip to content

Out of memory errors with some memory exceeding operations (Ex: OPR + mixed states) #99

Description

@ShantanuKodgirwar

Currently, there is no proper batching infrastructure, so the data cannot really be streamed in a GPU when there is smaller VRAM. For example: in the below screenshot when running with OPR and mixed states on a 32GB GPU, the memory is insufficient very often. This needs to be taken care of to make the same code GPU agnostic.

Image

Metadata

Metadata

Labels

bugSomething isn't workingenhancementNew feature or request

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions