saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/.github/workflows
History
Nikita Shulga 92ddb3d3d3 [MPS] Expose MPSProfiler::start/stopCapture to Python (#144561) 
I.e. when `MTL_CAPTURE_ENABLED` environment variable is set to 1, one should be able to invoke wrap the code with `torch.mps.profiler.capture_metal` to generate gputrace for shaders invoked inside the context manager.

For example, code below:
```python
import torch
import os

def foo(x):
   return x[:,::2].sin() + x[:, 1::2].cos()

if __name__ == "__main__":
    os.environ["MTL_CAPTURE_ENABLED"] = "1"
    x = torch.rand(32, 1024, device="mps")

    with torch.mps.profiler.metal_capture("compiled_shader"):
        torch.compile(foo)(x)
```
should capture the execution of a `torch.compile` generated shader
<img width="734" alt="image" src="https://github.com/user-attachments/assets/718ff64e-103b-4b11-b66c-c89cfc770b5d" />

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144561
Approved by: https://github.com/manuelcandales
ghstack dependencies: #144559, #144560
2025-01-11 02:05:36 +00:00
..
_bazel-build-test.yml
_binary-build-linux.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
_binary-test-linux.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
_binary-upload.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
_docs.yml
_linux-build.yml S390x cancelled jobs cleanup (#144149) 2025-01-09 20:45:19 +00:00
_linux-test.yml S390x ci periodic tests (#125401) 2025-01-10 18:21:07 +00:00
_mac-build.yml
_mac-test-mps.yml [MPS] Expose MPSProfiler::start/stopCapture to Python (#144561) 2025-01-11 02:05:36 +00:00
_mac-test.yml [Monitor] Enable non-perf linux test monitor (#142168) 2024-12-11 01:10:43 +00:00
_rocm-test.yml Use sccache 0.9.0 on ROCm build job (#144125) 2025-01-04 08:56:48 +00:00
_runner-determinator.yml Fix unused Python variables outside torch/ and test/ (#136359) 2024-12-11 17:10:23 +00:00
_win-build.yml
_win-test.yml
_xpu-test.yml
assigntome-docathon.yml
auto_request_review.yml
build-almalinux-images.yml
build-libtorch-images.yml [ROCm] upgrade nightly wheels to rocm6.3 - 1 of 2 (docker images) (#142151) 2024-12-13 16:21:17 +00:00
build-magma-linux.yml
build-magma-windows.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
build-manywheel-images-s390x.yml
build-manywheel-images.yml [ROCm] upgrade nightly wheels to rocm6.3 - 1 of 2 (docker images) (#142151) 2024-12-13 16:21:17 +00:00
build-triton-wheel.yml [ROCm] upgrade nightly wheels to rocm6.3 - 2 of 2 (binaries) (#143613) 2024-12-23 19:47:30 +00:00
check-labels.yml
check_mergeability_ghstack.yml
cherry-pick.yml
close-nonexistent-disable-issues.yml
create_release.yml [BE] Get rid of malfet/checkout@silent-checkout (#143516) 2024-12-19 00:36:36 +00:00
delete_old_branches.yml
docathon-sync-label.yml
docker-builds.yml Update inductor jobs to use CUDA 12.4 (#142177) 2024-12-09 16:18:38 +00:00
docker-release.yml Use validate-docker-images workflow from test-infra (#143081) 2024-12-12 00:24:27 +00:00
generated-linux-aarch64-binary-manywheel-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-binary-libtorch-cxx11-abi-main.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-binary-libtorch-cxx11-abi-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-binary-libtorch-pre-cxx11-main.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-binary-libtorch-pre-cxx11-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-binary-manywheel-main.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-binary-manywheel-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-linux-s390x-binary-manywheel-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-macos-arm64-binary-libtorch-cxx11-abi-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-macos-arm64-binary-wheel-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-windows-binary-libtorch-debug-main.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-windows-binary-libtorch-debug-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-windows-binary-libtorch-release-main.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-windows-binary-libtorch-release-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
generated-windows-binary-wheel-nightly.yml Remove builder repo from workflows and scripts (#143776) 2024-12-24 14:11:51 +00:00
inductor-micro-benchmark-x86.yml Uniformly pass secrets: inherit to all jobs that go to _linux-build/_linux-test (#141995) 2024-12-05 14:52:43 +00:00
inductor-micro-benchmark.yml Migrate the rest of CUDA 12.1 jobs to 12.4 (#144118) 2025-01-03 17:45:41 +00:00
inductor-perf-compare.yml Migrate the rest of CUDA 12.1 jobs to 12.4 (#144118) 2025-01-03 17:45:41 +00:00
inductor-perf-test-nightly-aarch64.yml [Monitor] Enable non-perf linux test monitor (#142168) 2024-12-11 01:10:43 +00:00
inductor-perf-test-nightly-macos.yml [Monitor] Enable non-perf linux test monitor (#142168) 2024-12-11 01:10:43 +00:00
inductor-perf-test-nightly-x86.yml [Monitor] Enable non-perf linux test monitor (#142168) 2024-12-11 01:10:43 +00:00
inductor-perf-test-nightly.yml Inductor dashboard benchmarks: swap unused freeze_autotune_cudagraphs workflow for cppwrapper workflow (#144427) 2025-01-09 23:56:00 +00:00
inductor-periodic.yml Migrate the rest of CUDA 12.1 jobs to 12.4 (#144118) 2025-01-03 17:45:41 +00:00
inductor-rocm.yml [ROCm][CI] upgrade CI to ROCm 6.3 (#142152) 2025-01-09 17:14:16 +00:00
inductor-unittest.yml ir.ExternKernel: correctly handle kwarg default arguments (#141371) 2025-01-03 16:05:31 +00:00
inductor.yml Update inductor jobs to use CUDA 12.4 (#142177) 2024-12-09 16:18:38 +00:00
lint-autoformat.yml
lint-bc.yml
lint.yml Fix missing tests on test tool lint job (#143052) 2024-12-12 20:29:32 +00:00
linux-aarch64.yml [CI] Run aarch64 tests on Graviton3 (#143129) 2024-12-13 07:39:22 +00:00
llm_td_retrieval.yml
mac-mps.yml Uniformly pass secrets: inherit to all jobs that go to _linux-build/_linux-test (#141995) 2024-12-05 14:52:43 +00:00
nightly-s3-uploads.yml Some workflows to use oidc instead of AWS keys (#142264) 2024-12-10 19:40:23 +00:00
nightly.yml Uniformly pass secrets: inherit to all jobs that go to _linux-build/_linux-test (#141995) 2024-12-05 14:52:43 +00:00
nitpicker.yml
periodic.yml [ROCm][CI] upgrade CI to ROCm 6.3 (#142152) 2025-01-09 17:14:16 +00:00
pull.yml [ROCm][CI] upgrade CI to ROCm 6.3 (#142152) 2025-01-09 17:14:16 +00:00
revert.yml
rocm.yml [ROCm][CI] upgrade CI to ROCm 6.3 (#142152) 2025-01-09 17:14:16 +00:00
runner-determinator-validator.yml
runner_determinator_script_sync.yaml
s390.yml Uniformly pass secrets: inherit to all jobs that go to _linux-build/_linux-test (#141995) 2024-12-05 14:52:43 +00:00
s390x-periodic.yml S390x ci periodic tests (#125401) 2025-01-10 18:21:07 +00:00
scorecards.yml
slow.yml [ROCm][CI] upgrade CI to ROCm 6.3 (#142152) 2025-01-09 17:14:16 +00:00
stale.yml
target-determination-indexer.yml
target_determination.yml
test-check-binary.yml Add check_binary workflow to pytorch/pytorch (#143201) 2024-12-13 19:30:10 +00:00
torchbench.yml Uniformly pass secrets: inherit to all jobs that go to _linux-build/_linux-test (#141995) 2024-12-05 14:52:43 +00:00
trunk.yml [ROCm][CI] upgrade CI to ROCm 6.3 (#142152) 2025-01-09 17:14:16 +00:00
trymerge.yml
tryrebase.yml
unstable-periodic.yml
unstable.yml
update-viablestrict.yml
update_pytorch_labels.yml
upload-test-stats-while-running.yml
upload-test-stats.yml
upload-torch-dynamo-perf-stats.yml Add macos perf run to the dashboard upload (#141999) 2024-12-04 01:08:13 +00:00
upload_test_stats_intermediate.yml Some workflows to use oidc instead of AWS keys (#142264) 2024-12-10 19:40:23 +00:00
weekly.yml
xpu.yml Uniformly pass secrets: inherit to all jobs that go to _linux-build/_linux-test (#141995) 2024-12-05 14:52:43 +00:00