Enable cpu offload with weights inside the module #3602

	name: Run Tests

	on:
	pull_request:
	paths:
	- "src/**"
	- "tests/**"
	- ".github/**"
	- "examples/**"
	- "setup.py"
	types: [opened, synchronize, reopened]

	env:
	HF_HOME: ~/hf_cache
	TESTING_MOCKED_DATALOADERS: "1"
	IS_GITHUB_CI: "1"

	jobs:
	run-tests:
	runs-on: ubuntu-latest
	strategy:
	fail-fast: false
	matrix:
	pytorch-version: [
	latest,
	minimum,
	]
	test-kind: [
	test_prod,
	test_core,
	test_cli,
	test_big_modeling,
	test_deepspeed,
	test_fsdp,
	test_example_differences,
	test_checkpoint_step,
	test_checkpoint_epoch,
	test_rest
	]
	steps:
	- uses: actions/[email protected]
	- name: Set up python 3.8
	uses: actions/setup-python@v3
	with:
	python-version: 3.8

	- name: Activate python cache
	uses: actions/cache@v3
	with:
	path: \|
	${{ env.pythonLocation }}
	${{ env.HF_HOME }}
	key: ${{ env.pythonLocation }}-${{ matrix.pytorch-version }}-${{ matrix.test-kind }}-${{ hashFiles('setup.py') }}

	- name: Install the library
	run: \|
	pip install --upgrade pip
	if [[ ${{ matrix.test-kind }} = test_prod ]]; then pip install -e .[test_prod]; fi
	if [[ ${{ matrix.test-kind }} != test_prod ]]; then pip install -e .[testing,test_trackers]; fi
	if [[ ${{ matrix.test-kind }} = test_rest ]]; then pip uninstall comet_ml -y; fi
	if [[ ${{ matrix.test-kind }} = minimum ]]; then pip install torch==1.10.0; fi
	pip install pytest-reportlog tabulate

	- name: Run Tests
	env:
	PYTORCH_VERSION: ${{ matrix.pytorch-version }}
	run: \|
	make ${{ matrix.test-kind }}

	- name: Generate Report
	if: always()
	run: \|
	python utils/log_reports.py >> $GITHUB_STEP_SUMMARY

Provide feedback