File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -21,9 +21,6 @@ torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
2121torchvision==0.22.0; platform_machine == "ppc64le"
2222datasets # for benchmark scripts
2323
24- # cpu cannot use triton 3.3.0
25- triton==3.2.0; platform_machine == "x86_64"
26-
2724# Intel Extension for PyTorch, only for x86_64 CPUs
2825intel-openmp==2024.2.1; platform_machine == "x86_64"
2926intel_extension_for_pytorch==2.7.0; platform_machine == "x86_64"
Original file line number Diff line number Diff line change @@ -68,9 +68,11 @@ class TritonPlaceholder(types.ModuleType):
6868
6969 def __init__ (self ):
7070 super ().__init__ ("triton" )
71+ self .__version__ = "3.3.0"
7172 self .jit = self ._dummy_decorator ("jit" )
7273 self .autotune = self ._dummy_decorator ("autotune" )
7374 self .heuristics = self ._dummy_decorator ("heuristics" )
75+ self .Config = self ._dummy_decorator ("Config" )
7476 self .language = TritonLanguagePlaceholder ()
7577 logger .warning_once (
7678 "Triton is not installed. Using dummy decorators. "
You can’t perform that action at this time.
0 commit comments