triton/python/triton/__init__.py

"""isort:skip_file"""
# flake8: noqa: F401
__version__ = '2.0.0'

# TODO: torch needs to be imported first
# or pybind11 shows `munmap_chunk(): invalid pointer`
import torch
# submodules
from .code_gen import cdiv, next_power_of_2, jit, autotune, heuristics, \
    JITFunction, Config, Autotuner, reinterpret
from . import language
from . import code_gen
from . import testing
from . import ops
[STYLE] add isort and autopep8 config files and check on CI (#423) Also a fix a few more style issues from the "aggressive" mode of autopep8. 2022-01-07 13:11:34 -08:00			`"""isort:skip_file"""`
[STYLE] check python with flake8 (#424) I've been using this locally to find errors without running tests, and now that we're using autopep8, it passes with minimal suppressions. This is also what turned up the issues with the tutorials, which were fixed in #422. 2022-01-07 15:28:36 -08:00			`# flake8: noqa: F401`
[VERSION] Bumped triton.__version__ to 2.0.0 2021-11-12 15:10:04 -08:00			`__version__ = '2.0.0'`
[FRONTEND] Add cache_version to triton.jit (#301) 2021-09-23 16:45:54 -07:00
[PYTHON][TESTS][DOC] Various improvement of the API and code quality: * Simplified `triton.kernel` API to achieve lower latency: > .data_ptr() must now be passed as kernel argument. No more implicit conversion from torch.tensor > compilation options are now constant attributes, i.e., opt.d('VAR') becomes opt.VAR > torch.device must now be passed explicitly to triton.kernel (no longer inferred from torch.tensor arguments) * C++ tests moved to `python/tests/` * C++ tutorial created in `tutorials/` * Python tutorial created in python/tutorials/ * Version changed to 1.0alpha * No longer copying C++ headers into the Python package * added python/triton/ops/ package for pre-written Triton ops 2021-01-29 17:27:16 -05:00			`# TODO: torch needs to be imported first`
			# or pybind11 shows `munmap_chunk(): invalid pointer`
			`import torch`
[PYTHON] Some cleaning of the PyBind11 wrappers (#62) 2021-02-06 17:10:44 -08:00			`# submodules`
[FRONTEND] Semantic analysis refactor (#491) Moved dispatch.cc to semantic.py (@ptillet) Integer signedness analysis was moved from C++ to python (@daadaada) Cleaner frontend types (@daadaada) Moved SSA construction to a separate object (@ptillet) Co-authored-by: Yan Da <dyanab@connect.ust.hk> 2022-04-06 16:13:53 -07:00			`from .code_gen import cdiv, next_power_of_2, jit, autotune, heuristics, \`
			`JITFunction, Config, Autotuner, reinterpret`
[PYTHON] Renamed triton.core -> triton.language (#92) 2021-04-23 17:18:14 -04:00			`from . import language`
[DOCS] Updates and improvements (#87) 2021-04-22 10:27:02 -04:00			`from . import code_gen`
[PYTHON] Added automated benchmark script (#63) This adds a bench functionality to the setup.py that can be used to run the benchmark suite and generates a bunch of csv files (and optionally plots) python setup.py bench python setup.py bench --with-plots python setup.py bench --filter=cross_entropy 2021-02-08 12:16:41 -08:00			`from . import testing`
[PYTHON] Added triton.__version__ string 2021-02-03 17:38:48 -05:00			`from . import ops`