pytorch
[WIP] Add dtype attribute to TritonCSEVariable
Summary:
-
This diff introduces
dtypeattribute toTritonCSEVariableand a dtype propagation helper function to infer dtype from input to output for each op. -
There will be a follow-up diff that uses this
dtypeinformation inTritonCSEVariableto perform dtype-aware codegen.
Test Plan: CI
Differential Revision: D61815079
cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang
:link: Helpful Links
:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/136778
- :page_facing_up: Preview Python docs built from this PR
- :page_facing_up: Preview C++ docs built from this PR
- :question: Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours
Note: Links to docs will display an error until the docs builds have been completed.
:x: 4 New Failures, 3 Unrelated Failures
As of commit ea0a9e69f0d81f9f9f6ba35034b81f5d059f7ccd with merge base 245026af2d2f26c74993cb90e01bddbd627c6797 (
NEW FAILURES - The following jobs have failed:
-
pull / linux-focal-cuda12.1-py3.10-gcc9-bazel-test / build-and-test (default, 1, 1, linux.4xlarge.nvidia.gpu) (gh)
Build completed, 1 test FAILED, 3216 total actions -
trunk / macos-py3-arm64 / test (default, 1, 3, macos-m1-stable) (gh)
RuntimeError: inductor/test_max_autotune 1/1 failed! -
trunk / macos-py3-arm64 / test (default, 2, 3, macos-m1-stable) (gh)
RuntimeError: inductor/test_torchinductor 1/2 failed! -
trunk / macos-py3-arm64 / test (default, 3, 3, macos-m1-stable) (gh)
RuntimeError: inductor/test_torchinductor 2/2 failed!
BROKEN TRUNK - The following jobs failed but were present on the merge base:
👉 Rebase onto the `viable/strict` branch to avoid these failures
-
pull / linux-focal-cuda12.4-py3.10-gcc9-bazel-test / build-and-test (default, 1, 1, linux.4xlarge.nvidia.gpu) (gh) (trunk failure)
Build completed, 1 test FAILED, 3216 total actions -
trunk / macos-py3-arm64-mps / test (mps, 1, 1, macos-m1-13) (gh) (trunk failure)
RuntimeError: test_mps 1/11 failed! -
trunk / macos-py3-arm64-mps / test (mps, 1, 1, macos-m1-14) (gh) (trunk failure)
RuntimeError: test_mps 1/11 failed!
This comment was automatically generated by Dr. CI and updates every 15 minutes.
![pytorch-bot[bot] avatar](https://avatars.githubusercontent.com/in/40112?v=4 "Published by a pub.dev verified publisher"){kind=small}
