4bcf6fdd93291a1bfba3c7038fe30a7338870968,onmt/models/SRU.py,,,#,72

Before Change

    device = torch.device("cuda")
    tmp_ = torch.rand(1, 1).to(device)

    sru_prog = Program(SRU_CODE.encode("utf-8"),
                       "sru_prog.cu".encode("utf-8"))
    sru_ptx = sru_prog.compile()
    sru_mod = function.Module()
    sru_mod.load(bytes(sru_ptx.encode()))

    SRU_FWD_FUNC = sru_mod.get_function("sru_fwd")

After Change

}

SRU_FWD_FUNC, SRU_BWD_FUNC = None, None
SRU_BiFWD_FUNC, SRU_BiBWD_FUNC = None, None
SRU_STREAM = None

Instances