init
This commit is contained in:
parent
baa9bc407b
commit
e14b3cfbbe
1 changed files with 0 additions and 1 deletions
|
@ -35,7 +35,6 @@ def main():
|
||||||
|
|
||||||
# CUDA Graphs to reduce launch overhead when shapes are static
|
# CUDA Graphs to reduce launch overhead when shapes are static
|
||||||
graph = None
|
graph = None
|
||||||
s_out = torch.zeros((), dtype=torch.float32, device=device)
|
|
||||||
if device.type == "cuda":
|
if device.type == "cuda":
|
||||||
stream = torch.cuda.Stream()
|
stream = torch.cuda.Stream()
|
||||||
torch.cuda.synchronize()
|
torch.cuda.synchronize()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue