Add UNKNOWN PipelineState type (meta-pytorch#3529)

Summary:
Pull Request resolved: meta-pytorch#3529

To use in next diff.

Reviewed By: guoding83128

Differential Revision: D86636523

fbshipit-source-id: bc46bf5d8899999d51ba68bf9dc576954d38b400

Author: ergaziz

torchrec/distributed/train_pipeline/types.py

@@ -96,6 +96,7 @@ class PipelineState(Enum):
     IDLE = 0
     CALL_FWD = 1
     CALL_BWD = 2
+    UNKNOWN = 3
 
     def __str__(self) -> str:
         return self.name