mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
2c0d607882
Fixes #101469 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101475 Approved by: https://github.com/ezyang
31 lines
894 B
Python
31 lines
894 B
Python
# Owner(s): ["module: bazel"]
|
|
|
|
"""
|
|
This test module contains a minimalistic "smoke tests" for the bazel build.
|
|
|
|
Currently it doesn't use any testing framework (i.e. pytest)
|
|
TODO: integrate this into the existing pytorch testing framework.
|
|
|
|
The name uses underscore `_test_bazel.py` to avoid globbing into other non-bazel configurations.
|
|
"""
|
|
|
|
import torch
|
|
|
|
def test_sum() -> None:
|
|
assert torch.eq(torch.tensor([[1, 2, 3]]) + torch.tensor([[4, 5, 6]]), torch.tensor([[5, 7, 9]])).all()
|
|
|
|
def test_simple_compile_eager() -> None:
|
|
|
|
def foo(x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
|
|
a = torch.sin(x)
|
|
b = torch.cos(y)
|
|
return a + b
|
|
|
|
opt_foo1 = torch.compile(foo, backend="eager")
|
|
# just check that we can run without raising an Exception
|
|
assert opt_foo1(torch.randn(10, 10), torch.randn(10, 10)) is not None
|
|
|
|
|
|
test_sum()
|
|
test_simple_compile_eager()
|