[MLIR][KERNEL_GEN] Add a library to lower kernels with the host side.
* Unified TF->Cubin and TF->Kernel_with_host side lowering in `kernel_creator.h|cc` * Added a pass that attaches GPU binary blob to GPUModuleOp * Refactored most of the code. * Added tf_to_kernel binary that emits obj file PiperOrigin-RevId: 330494488
This commit is contained in:
		
							parent
							
								
									64d9a465ee
								
							
						
					
					
						commit
						ebc7992d31
					
				|  | @ -53,7 +53,7 @@ std::unique_ptr<OperationPass<ModuleOp>> createLegalizeToLhloPass( | |||
| std::unique_ptr<OperationPass<FuncOp>> createLegalizeHloToLinalgPass(); | ||||
| 
 | ||||
| // Transforms unranked HLO operations to ranked ones where possible.
 | ||||
| std::unique_ptr<OperationPass<FuncOp>> createTransformUnrankedHloPass(); | ||||
| std::unique_ptr<FunctionPass> createTransformUnrankedHloPass(); | ||||
| 
 | ||||
| // Sinks constants implicitly captured in control flow regions. This is
 | ||||
| // necessary to export to XLA.
 | ||||
|  |  | |||
|  | @ -198,7 +198,7 @@ void PopulateTransformUnrankedHloPatterns(MLIRContext *context, | |||
|   // clang-format on
 | ||||
| } | ||||
| 
 | ||||
| std::unique_ptr<::mlir::Pass> createTransformUnrankedHloPass() { | ||||
| std::unique_ptr<FunctionPass> createTransformUnrankedHloPass() { | ||||
|   return std::make_unique<TransformUnrankedHloPass>(); | ||||
| } | ||||
| 
 | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue