diff --git a/op_builder/builder.py b/op_builder/builder.py index 09b781fddd72..40c37df979f4 100644 --- a/op_builder/builder.py +++ b/op_builder/builder.py @@ -407,18 +407,6 @@ def simd_width(self): return '-D__AVX256__' return '-D__SCALAR__' - def python_requirements(self): - ''' - Override if op wants to define special dependencies, otherwise will - take self.name and load requirements-.txt if it exists. - ''' - path = f'requirements/requirements-{self.name}.txt' - requirements = [] - if os.path.isfile(path): - with open(path, 'r') as fd: - requirements = [r.strip() for r in fd.readlines()] - return requirements - def command_exists(self, cmd): if '|' in cmd: cmds = cmd.split("|") diff --git a/setup.py b/setup.py index 24e5ec62b7dd..969060018e7c 100755 --- a/setup.py +++ b/setup.py @@ -60,7 +60,8 @@ def fetch_requirements(path): 'autotuning': fetch_requirements('requirements/requirements-autotuning.txt'), 'autotuning_ml': fetch_requirements('requirements/requirements-autotuning-ml.txt'), 'sparse_attn': fetch_requirements('requirements/requirements-sparse_attn.txt'), - 'inf': fetch_requirements('requirements/requirements-inf.txt') + 'inf': fetch_requirements('requirements/requirements-inf.txt'), + 'sd': fetch_requirements('requirements/requirements-sd.txt') } # Add specific cupy version to both onebit extension variants @@ -150,11 +151,6 @@ def op_enabled(op_name): builder.warning(f"One can disable {op_name} with {env_var}=0") abort(f"Unable to pre-compile {op_name}") - # If op is compatible update install reqs so it can potentially build/run later - if op_compatible: - reqs = builder.python_requirements() - install_requires += builder.python_requirements() - # if op is compatible but install is not enabled (JIT mode) if is_rocm_pytorch and op_compatible and not op_enabled(op_name): builder.hipify_extension()