How to compute loss between tensor and LazyTensor

Hi I am using a package called koila. They claimed that using their package can save from CUDA of out memory error but when I tried using it, it is giving UnsupportedType error when I am trying to compute CrossEntropyLoss.