2023-12-21 07:35:15 +00:00
|
|
|
CONDA_ENV_PATH = "transformers.yml"
|
|
|
|
|
|
|
|
ifeq ($(BUILD_TYPE), cublas)
|
|
|
|
CONDA_ENV_PATH = "transformers-nvidia.yml"
|
|
|
|
endif
|
|
|
|
|
2024-02-16 14:08:50 +00:00
|
|
|
ifeq ($(BUILD_TYPE), hipblas)
|
|
|
|
CONDA_ENV_PATH = "transformers-rocm.yml"
|
|
|
|
endif
|
|
|
|
|
2024-03-07 13:37:45 +00:00
|
|
|
# Intel GPU are supposed to have dependencies installed in the main python
|
|
|
|
# environment, so we skip conda installation for SYCL builds.
|
|
|
|
# https://github.com/intel/intel-extension-for-pytorch/issues/538
|
|
|
|
ifneq (,$(findstring sycl,$(BUILD_TYPE)))
|
|
|
|
export SKIP_CONDA=1
|
|
|
|
endif
|
|
|
|
|
2023-12-21 07:35:15 +00:00
|
|
|
.PHONY: transformers
|
|
|
|
transformers:
|
|
|
|
@echo "Installing $(CONDA_ENV_PATH)..."
|
|
|
|
bash install.sh $(CONDA_ENV_PATH)
|