[mlir][spirv] NFC: split deserialization into multiple source files

This avoids large source files and gives a better structure. It also
allows leveraging compilation parallelism.

Reviewed By: mravishankar

Differential Revision: https://reviews.llvm.org/D94360

Author: antiagainst

mlir/lib/Target/CMakeLists.txt

@@ -1,3 +1,5 @@
+add_subdirectory(SPIRV)
+
 add_mlir_translation_library(MLIRTargetLLVMIRModuleTranslation
   LLVMIR/DebugTranslation.cpp
   LLVMIR/ModuleTranslation.cpp
@@ -132,52 +134,3 @@ add_mlir_translation_library(MLIRTargetROCDLIR
   MLIRROCDLIR
   MLIRTargetLLVMIRModuleTranslation
   )
-
-add_mlir_translation_library(MLIRSPIRVBinaryUtils
-  SPIRV/SPIRVBinaryUtils.cpp
-
-  LINK_LIBS PUBLIC
-  MLIRIR
-  MLIRSPIRV
-  MLIRSupport
-  )
-
-add_mlir_translation_library(MLIRSPIRVSerialization
-  SPIRV/Serialization.cpp
-
-  DEPENDS
-  MLIRSPIRVSerializationGen
-
-  LINK_LIBS PUBLIC
-  MLIRIR
-  MLIRSPIRV
-  MLIRSPIRVBinaryUtils
-  MLIRSupport
-  MLIRTranslation
-  )
-
-add_mlir_translation_library(MLIRSPIRVDeserialization
-  SPIRV/Deserialization.cpp
-
-  DEPENDS
-  MLIRSPIRVSerializationGen
-
-  LINK_LIBS PUBLIC
-  MLIRIR
-  MLIRSPIRV
-  MLIRSPIRVBinaryUtils
-  MLIRSupport
-  MLIRTranslation
-  )
-
-add_mlir_translation_library(MLIRSPIRVTranslateRegistration
-  SPIRV/TranslateRegistration.cpp
-
-  LINK_LIBS PUBLIC
-  MLIRIR
-  MLIRSPIRV
-  MLIRSPIRVSerialization
-  MLIRSPIRVDeserialization
-  MLIRSupport
-  MLIRTranslation
-  )

mlir/lib/Target/SPIRV/CMakeLists.txt

@@ -0,0 +1,28 @@
+add_subdirectory(Deserialization)
+add_subdirectory(Serialization)
+
+set(LLVM_OPTIONAL_SOURCES
+  SPIRVBinaryUtils.cpp
+  TranslateRegistration.cpp
+  )
+
+add_mlir_translation_library(MLIRSPIRVBinaryUtils
+  SPIRVBinaryUtils.cpp
+
+  LINK_LIBS PUBLIC
+  MLIRIR
+  MLIRSPIRV
+  MLIRSupport
+  )
+
+add_mlir_translation_library(MLIRSPIRVTranslateRegistration
+  TranslateRegistration.cpp
+
+  LINK_LIBS PUBLIC
+  MLIRIR
+  MLIRSPIRV
+  MLIRSPIRVSerialization
+  MLIRSPIRVDeserialization
+  MLIRSupport
+  MLIRTranslation
+  )

mlir/lib/Target/SPIRV/Deserialization/CMakeLists.txt

@@ -0,0 +1,17 @@
+add_mlir_translation_library(MLIRSPIRVDeserialization
+  DeserializeOps.cpp
+  Deserializer.cpp
+  Deserialization.cpp
+
+  DEPENDS
+  MLIRSPIRVSerializationGen
+
+  LINK_LIBS PUBLIC
+  MLIRIR
+  MLIRSPIRV
+  MLIRSPIRVBinaryUtils
+  MLIRSupport
+  MLIRTranslation
+  )
+
+

mlir/lib/Target/SPIRV/Deserialization/Deserialization.cpp

@@ -0,0 +1,23 @@
+//===- Deserialization.cpp - MLIR SPIR-V Deserialization ------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "mlir/Target/SPIRV/Deserialization.h"
+
+#include "Deserializer.h"
+
+namespace mlir {
+spirv::OwningSPIRVModuleRef spirv::deserialize(ArrayRef<uint32_t> binary,
+                                               MLIRContext *context) {
+  Deserializer deserializer(binary, context);
+
+  if (failed(deserializer.deserialize()))
+    return nullptr;
+
+  return deserializer.collect();
+}
+} // namespace mlir