[Input] Register IREETensorExtDialect for Torch plugin (#22719)

Registers `IREETensorExtDialect` to fix unregistered dialect error from
bitcast ops created by BitCastTensorPass. Strangely, this error only
occurs when printing IR and debug info, which is why it's not getting
caught by CI. I think this was missed when `BitCastOp` was moved out of
Flow.

Signed-off-by: Ian Wood <[email protected]>

Author: IanWood1

compiler/plugins/input/Torch/PluginRegistration.cpp

@@ -9,6 +9,7 @@
 #include "iree/compiler/Dialect/HAL/IR/HALDialect.h"
 #include "iree/compiler/Dialect/LinalgExt/IR/LinalgExtDialect.h"
 #include "iree/compiler/Dialect/Stream/IR/StreamDialect.h"
+#include "iree/compiler/Dialect/TensorExt/IR/TensorExtDialect.h"
 #include "iree/compiler/Dialect/Util/IR/UtilDialect.h"
 #include "iree/compiler/PluginAPI/Client.h"
 #include "mlir/Dialect/MLProgram/IR/MLProgram.h"
@@ -66,6 +67,7 @@ struct TorchSession
     registry.insert<IREE::Flow::FlowDialect>();
     registry.insert<IREE::HAL::HALDialect>();
     registry.insert<IREE::Stream::StreamDialect>();
+    registry.insert<IREE::TensorExt::IREETensorExtDialect>();
     registry.insert<IREE::Util::UtilDialect>();
   }