ROCM illegal l memory access GPU hang on a second generation up refinement stage

#107
by ibyteohdear - opened

Greets,

With the latest packages lora fusing is much better on the whole, so that is cool however, on ROCM i needed the following to stop illegal memory access GPU hang on a second generation up refinement stage just fyi.

#patch sencond generation run on ROCM
import ltx_core.devices as _devices
import ltx_core.loader.lora as _lora

if torch.version.hip is not None:
_sync = lambda device: False
_devices.allow_async_transfer = _sync
_lora.allow_async_transfer = _sync

Sign up or log in to comment