Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions op_builder/builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,7 +296,7 @@ def cxx_args(self):
'''
return []

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
'''
Check if all non-python dependencies are satisfied to build this op
'''
Expand Down Expand Up @@ -679,7 +679,7 @@ def version_dependent_macros(self):
version_ge_1_5 = ['-DVERSION_GE_1_5']
return version_ge_1_1 + version_ge_1_3 + version_ge_1_5

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
return super().is_compatible(verbose)

def builder(self):
Expand Down
4 changes: 2 additions & 2 deletions op_builder/cpu/comm.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def include_paths(self):
def cxx_args(self):
return ['-O2', '-fopenmp']

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
# TODO: add soft compatibility check for private binary release.
# a soft check, as in we know it can be trivially changed.
return super().is_compatible(verbose)
Expand Down Expand Up @@ -65,7 +65,7 @@ def include_paths(self):
def cxx_args(self):
return ['-O2', '-fopenmp']

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
# TODO: add soft compatibility check for private binary release.
# a soft check, as in we know it can be trivially changed.
return super().is_compatible(verbose)
2 changes: 1 addition & 1 deletion op_builder/inference_core_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ def __init__(self, name=None):
def absolute_name(self):
return f'deepspeed.inference.v2.kernels{self.NAME}'

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
try:
import torch
except ImportError:
Expand Down
2 changes: 1 addition & 1 deletion op_builder/inference_cutlass_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ def __init__(self, name=None):
def absolute_name(self):
return f'deepspeed.inference.v2.kernels.cutlass_ops.{self.NAME}'

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
try:
import torch
except ImportError:
Expand Down
2 changes: 1 addition & 1 deletion op_builder/npu/async_io.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def check_for_libaio_pkg(self):
break
return found

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
# Check for the existence of libaio by using distutils
# to compile and link a test program that calls io_submit,
# which is a function provided by libaio that is used in the async_io op.
Expand Down
2 changes: 1 addition & 1 deletion op_builder/ragged_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ def __init__(self, name=None):
def absolute_name(self):
return f'deepspeed.inference.v2.kernels.ragged_ops.{self.NAME}'

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
try:
import torch
except ImportError:
Expand Down
2 changes: 1 addition & 1 deletion op_builder/ragged_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ def __init__(self, name=None):
def absolute_name(self):
return f'deepspeed.inference.v2.{self.NAME}'

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
try:
import torch
except ImportError:
Expand Down
1 change: 0 additions & 1 deletion op_builder/sparse_attn.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,6 @@ def is_compatible(self, verbose=False):
if verbose:
self.warning(
f'{self.NAME} requires a torch version >= 1.5 and < 2.0 but detected {TORCH_MAJOR}.{TORCH_MINOR}')

try:
import triton
except ImportError:
Expand Down
2 changes: 1 addition & 1 deletion op_builder/spatial_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def __init__(self, name=None):
def absolute_name(self):
return f'deepspeed.ops.spatial.{self.NAME}_op'

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
try:
import torch
except ImportError:
Expand Down
2 changes: 1 addition & 1 deletion op_builder/transformer_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def __init__(self, name=None):
def absolute_name(self):
return f'deepspeed.ops.transformer.inference.{self.NAME}_op'

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
try:
import torch
except ImportError:
Expand Down
2 changes: 1 addition & 1 deletion op_builder/xpu/async_io.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ def check_for_libaio_pkg(self):
break
return found

def is_compatible(self, verbose=True):
def is_compatible(self, verbose=False):
# Check for the existence of libaio by using distutils
# to compile and link a test program that calls io_submit,
# which is a function provided by libaio that is used in the async_io op.
Expand Down