[iOS GPU] Add debug information to track memory allocation exception

[xta0]

Stack from ghstack:

• [iOS GPU] [BE] use channel-last to transform the weights #59113 [iOS GPU] [BE] use channel-last to transform the weights
• [iOS GPU] Add debug information to track memory allocation exception #59112 [iOS GPU] Add debug information to track memory allocation exception

Differential Revision: D28730604

[facebook-github-bot]

💊 CI failures summary and remediations

As of commit 07dd9cc (more details on the Dr. CI page):

---

• 1/2 failures possibly* introduced in this PR
  • 1/1 non-scanned failure(s)
• 1/2 tentatively recognized as flaky ❄️
  • Click here to rerun these jobs

---

❄️ 1 failure tentatively classified as flaky

but reruns have not yet been triggered to confirm:

pytorch_xla_linux_bionic_py3_6_clang9_test (1/1)

Step: "Run tests" (full log | diagnosis details | 🔁 rerun) ❄️

May 28 01:24:01 test_bool_indices_accumulate_...: 55536 Segmentation fault (core dumped) "$@"

May 28 01:23:55 + run_test python3 /var/lib/jenkins/workspace/xla/test/../../test/test_indexing.py -v TestIndexingXLA
May 28 01:23:55 + python3 /var/lib/jenkins/workspace/xla/test/../../test/test_indexing.py -v TestIndexingXLA
May 28 01:23:57 Test results will be stored in test-reports/python-unittest/.var.lib.jenkins.workspace.xla.test.......test.test_indexing
May 28 01:23:57 
May 28 01:23:57 Running tests...
May 28 01:23:57 ----------------------------------------------------------------------
May 28 01:23:57   test_advancedindex_big_xla (__main__.TestIndexingXLA) ... ok (0.159s)
May 28 01:23:57   test_advancedindex_xla_float16 (__main__.TestIndexingXLA) ... skip (0.037s)
May 28 01:23:57   test_advancedindex_xla_float64 (__main__.TestIndexingXLA) ... skip (0.035s)
May 28 01:23:57   test_basic_advanced_combined_xla (__main__.TestIndexingXLA) ... ok (0.102s)
May 28 01:24:01   test_bool_indices_accumulate_xla (__main__.TestIndexingXLA) ... ./test/run_tests.sh: line 35: 55536 Segmentation fault      (core dumped) "$@"
May 28 01:24:01 + cleanup
May 28 01:24:01 + retcode=139
May 28 01:24:01 + set +x
May 28 01:24:01 =================== sccache compilation log ===================
May 28 01:24:01 =========== If your build fails, please take a look at the log above for possible reasons ===========
May 28 01:24:01 Compile requests                      28
May 28 01:24:01 Compile requests executed             26
May 28 01:24:01 Cache hits                             2
May 28 01:24:01 Cache hits (C/C++)                     2
May 28 01:24:01 Cache misses                          24

---

ci.pytorch.org: 1 failed

• Failed: pr/pytorch-linux-bionic-rocm4.2-py3.6

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

[facebook-github-bot]

This pull request has been merged in ab372ba.