Code

Check CUDA out of memory#213

Code

Draft

thodkatz wants to merge 5 commits intoilastik:mainilastik/tiktorch:mainfrom thodkatz:check-cuda-out-of-memorythodkatz/tiktorch:check-cuda-out-of-memoryCopy head branch name to clipboard

+764-585

Commits

Commits on Aug 19, 2024

• Remove unused proto buff of InputShape and OutputShape

  Show description for 2a7eb7a
  thodkatz

  committed
  2a7eb7a

• Move InputTensorValidator to converters module

  thodkatz

  committed
  529243e

• Add test data for emulating gpu out of memory

  thodkatz

  committed
  6f441ea

• Add procedures for checking gpu out of memory for given shapes

  Show description for b1595d7
  thodkatz

  committed
  b1595d7

• Add device id to cuda requests

  Show description for 46ac782
  thodkatz

  committed
  46ac782