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"
21
21
torchvision==0.22.0; platform_machine == "ppc64le"
22
22
datasets # for benchmark scripts
23
23
24
- # cpu cannot use triton 3.3.0
25
- triton==3.2.0; platform_machine == "x86_64"
26
-
27
24
# Intel Extension for PyTorch, only for x86_64 CPUs
28
25
intel-openmp==2024.2.1; platform_machine == "x86_64"
29
26
intel_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):
68
68
69
69
def __init__ (self ):
70
70
super ().__init__ ("triton" )
71
+ self .__version__ = "3.3.0"
71
72
self .jit = self ._dummy_decorator ("jit" )
72
73
self .autotune = self ._dummy_decorator ("autotune" )
73
74
self .heuristics = self ._dummy_decorator ("heuristics" )
75
+ self .Config = self ._dummy_decorator ("Config" )
74
76
self .language = TritonLanguagePlaceholder ()
75
77
logger .warning_once (
76
78
"Triton is not installed. Using dummy decorators. "
You can’t perform that action at this time.
0 commit comments