Fix 2.5.1 failing sparsity test (#1261) #835

	# From https://github.com/pytorch/test-infra/wiki/Using-Nova-Reusable-Build-Workflows
	name: Build Linux Wheels

	on:
	pull_request:
	paths:
	- build/packaging/**
	- .github/workflows/build_wheels_linux.yml
	- setup.py
	push:
	branches:
	- nightly
	- main
	- release/*
	tags:
	# NOTE: Binary build pipelines should only get triggered on release candidate builds
	# Release candidate tags look like: v1.11.0-rc1
	- v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
	schedule:
	- cron: '0 0 * * *' # Runs at midnight UTC every day
	workflow_dispatch:

	jobs:
	generate-matrix:
	uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
	with:
	package-type: wheel
	os: linux
	with-cpu: enable
	with-cuda: enable
	with-rocm: enable
	with-xpu: enable

	build:
	needs: generate-matrix
	permissions:
	id-token: write
	contents: read
	uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
	with:
	# Set the ref to an empty string instead of the default nightly because
	# torchao doesn't have nightly branch setup yet, instead the build is
	# triggered daily from main with a schedule
	repository: pytorch/ao
	ref: ""
	build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
	env-var-script: packaging/env_var_script_linux.sh
	pre-script: packaging/pre_build_script.sh
	post-script: packaging/post_build_script.sh
	smoke-test-script: packaging/smoke_test.py
	package-name: torchao
	trigger-event: ${{ github.event_name }}
	# This is the CUDA version to be uploaded to torchao-nightly pypi
	upload-to-pypi: cu121
	secrets:
	PYPI_API_TOKEN: ${{ secrets.PYPI_API_TOKEN }}

Provide feedback