chore: vendor sglang v0.5.10 snapshot
This commit is contained in:
14
third_party/sglang/sgl-kernel/tests/conftest.py
vendored
Normal file
14
third_party/sglang/sgl-kernel/tests/conftest.py
vendored
Normal file
@@ -0,0 +1,14 @@
|
||||
import pytest
|
||||
import torch
|
||||
|
||||
|
||||
# This fixture ensures the torch defaults don't get left in modified states between
|
||||
# tests (e.g., when a test fails before restoring the original value), which
|
||||
# can cause subsequent tests to fail.
|
||||
@pytest.fixture(autouse=True)
|
||||
def reset_torch_defaults():
|
||||
orig_default_device = torch.get_default_device()
|
||||
orig_default_dtype = torch.get_default_dtype()
|
||||
yield
|
||||
torch.set_default_dtype(orig_default_dtype)
|
||||
torch.set_default_device(orig_default_device)
|
||||
Reference in New Issue
Block a user