benchmark.py [21:31]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
options = parser.parse_args()

if options.example == 'py':
    from python.lltm import LLTM
elif options.example == 'cpp':
    from cpp.lltm import LLTM
else:
    from cuda.lltm import LLTM
    options.cuda = True

device = torch.device("cuda") if options.cuda else torch.device("cpu")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



grad_check.py [14:24]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
options = parser.parse_args()

if options.example == 'py':
    from python.lltm_baseline import LLTMFunction
elif options.example == 'cpp':
    from cpp.lltm import LLTMFunction
else:
    from cuda.lltm import LLTMFunction
    options.cuda = True

device = torch.device("cuda") if options.cuda else torch.device("cpu")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -