From 27c06ef29fb572ab01afe36df30651f65cd45e67 Mon Sep 17 00:00:00 2001
From: Phil26AT <phil.lindenberger@gmail.com>
Date: Tue, 16 Apr 2024 14:17:28 +0200
Subject: [PATCH] fix compilation for torch v2.2.1

---
 lightglue/lightglue.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lightglue/lightglue.py b/lightglue/lightglue.py
index 4d6b9d7..14e6a61 100644
--- a/lightglue/lightglue.py
+++ b/lightglue/lightglue.py
@@ -438,6 +438,7 @@ def compile(
                 stacklevel=2,
             )
 
+        torch._inductor.cudagraph_mark_step_begin()
         for i in range(self.conf.n_layers):
             self.transformers[i].masked_forward = torch.compile(
                 self.transformers[i].masked_forward, mode=mode, fullgraph=True