This repository was archived by the owner on Aug 1, 2025. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 129
This repository was archived by the owner on Aug 1, 2025. It is now read-only.
Smoke tests for dynamo/inductor/triton #1733
Copy link
Copy link
Closed
Description
Running pip3 install --pre torch[dynamo] --extra-index-url https://download.pytorch.org/whl/nightly/cu117
And then the below works
import torch
import torch._dynamo as dynamo
class MockModule(torch.nn.Module):
def __init__(self):
super().__init__()
self.relu = torch.nn.ReLU()
def forward(self, x):
return self.relu(torch.cos(x))
mod = MockModule()
optimized_mod = dynamo.optimize("inductor")(mod)
optimized_mod(torch.randn(10))
But then running python 01-vector-add.py
https://github.com/openai/triton/blob/master/python/tutorials/01-vector-add.py caused a
Traceback (most recent call last):
File "01-vector-add.py", line 17, in <module>
import triton
File "/home/nshulga/miniconda3/envs/py37-torch114-dynamo/lib/python3.7/site-packages/triton/__init__.py", line 10, in <module>
from .runtime import Config, autotune, heuristics, JITFunction, KernelInterface
File "/home/nshulga/miniconda3/envs/py37-torch114-dynamo/lib/python3.7/site-packages/triton/runtime/__init__.py", line 1, in <module>
from .autotuner import Config, Heuristics, autotune, heuristics # noqa: F401
File "/home/nshulga/miniconda3/envs/py37-torch114-dynamo/lib/python3.7/site-packages/triton/runtime/autotuner.py", line 7, in <module>
from ..testing import do_bench
File "/home/nshulga/miniconda3/envs/py37-torch114-dynamo/lib/python3.7/site-packages/triton/testing.py", line 9, in <module>
import triton._C.libtriton.triton as _triton
ImportError: /home/nshulga/miniconda3/envs/py37-torch114-dynamo/lib/python3.7/site-packages/triton/_C/libtriton.so: undefined symbol: _ZN4llvm14TargetRegistry12lookupTargetERKSsRSs
So we probably need some smoke tests @williamwen42 @malfet
Metadata
Metadata
Assignees
Labels
No labels