chore: vendor sglang v0.5.10 snapshot
This commit is contained in:
9
third_party/sglang/sgl-kernel/python/sgl_kernel/memory.py
vendored
Normal file
9
third_party/sglang/sgl-kernel/python/sgl_kernel/memory.py
vendored
Normal file
@@ -0,0 +1,9 @@
|
||||
import torch
|
||||
|
||||
|
||||
def weak_ref_tensor(tensor):
|
||||
return (
|
||||
torch.ops.sgl_kernel.weak_ref_tensor(tensor)
|
||||
if isinstance(tensor, torch.Tensor)
|
||||
else tensor
|
||||
)
|
||||
Reference in New Issue
Block a user