From 96c08688355fb5ab2dc0a32bac0775bc8ebbc570 Mon Sep 17 00:00:00 2001
From: Dominik Farhan <nikousf@seznam.cz>
Date: Mon, 1 Sep 2025 21:43:32 +0200
Subject: [PATCH] flops: add note on unsupported ops

---
 torchtnt/utils/flops.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/torchtnt/utils/flops.py b/torchtnt/utils/flops.py
index a7a404040e..aec8eb5737 100644
--- a/torchtnt/utils/flops.py
+++ b/torchtnt/utils/flops.py
@@ -179,6 +179,10 @@ class FlopTensorDispatchMode(TorchDispatchMode):
     Flop count implementation based on
     https://dev-discuss.pytorch.org/t/the-ideal-pytorch-flop-counter-with-torch-dispatch/505
 
+    Note that not all operators are supported.
+    Unsupported operations will be logged as debug messages but will not contribute
+    to the FLOP count.
+
     Examples::
 
         >>> import copy