Skip to content
This repository was archived by the owner on Aug 1, 2025. It is now read-only.
This repository was archived by the owner on Aug 1, 2025. It is now read-only.

Smoke tests for dynamo/inductor/triton #1733

@msaroufim

Description

@msaroufim

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
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions