Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
globals.py 56 Bytes
Newer Older
Nicolas Patry's avatar
Improving mamba runtime by using updates (#1552)  
Nicolas Patry committed
1
2
3 
import torch

MEM_POOL = torch.cuda.graph_pool_handle()