From 3fd601174571a62db907230164bde833454bdb9a Mon Sep 17 00:00:00 2001
From: Stonepia <tong.su@intel.com>
Date: Mon, 11 May 2026 23:29:16 -0700
Subject: [PATCH] Fix for intel/torch-xpu-ops#1969

torch._dynamo.exc.InternalTorchDynamoError: TypeError: cannot create weak reference to 'torch.Event' object
---
 torch/csrc/xpu/Event.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/torch/csrc/xpu/Event.cpp b/torch/csrc/xpu/Event.cpp
index eeacd0d30c624..db6725be4b1f1 100644
--- a/torch/csrc/xpu/Event.cpp
+++ b/torch/csrc/xpu/Event.cpp
@@ -45,6 +45,9 @@ static PyObject* THXPEvent_pynew(
 }
 
 static void THXPEvent_dealloc(THXPEvent* self) {
+  if (self->weakreflist != nullptr) {
+    PyObject_ClearWeakRefs((PyObject*)self);
+  }
   {
     pybind11::gil_scoped_release no_gil{};
     self->xpu_event.~XPUEvent();
@@ -168,7 +171,7 @@ static PyTypeObject THXPEventType = {
     nullptr, /* tp_traverse */
     nullptr, /* tp_clear */
     nullptr, /* tp_richcompare */
-    0, /* tp_weaklistoffset */
+    offsetof(THXPEvent, weakreflist), /* tp_weaklistoffset */
     nullptr, /* tp_iter */
     nullptr, /* tp_iternext */
     THXPEvent_methods, /* tp_methods */