lib/accy/src/preparation/root.zig

daab053ee43316e1809a84551d573ddd1e5bf3d2

  1 pub const backend = @import("backend.zig");
  2 pub const bufferization = @import("bufferization/root.zig");
  3 pub const call = @import("call.zig");
  4 pub const cache = @import("cache.zig");
  5 pub const canonicalization = @import("canonicalization.zig");
  6 pub const saturation = @import("saturation.zig");
  7 pub const dtype = @import("dtype.zig");
  8 pub const activation = @import("activation.zig");
  9 pub const einsum = @import("einsum.zig");
 10 pub const capture = @import("capture.zig");
 11 pub const folding = @import("folding.zig");
 12 pub const fusion = @import("fusion/root.zig");
 13 pub const indexing = @import("indexing.zig");
 14 pub const kernelization = @import("kernelization/root.zig");
 15 pub const layout = @import("layout.zig");
 16 pub const memory = @import("memory.zig");
 17 pub const outlining = @import("outlining/root.zig");
 18 pub const publication = @import("publication.zig");
 19 pub const recipe = @import("recipe.zig");
 20 pub const pipeline = @import("pipeline.zig");
 21 pub const schedule = @import("schedule/root.zig");
 22 pub const shape = @import("shape/root.zig");
 23 pub const target = @import("target.zig");
 24 
 25 pub const contract_pipeline_name = pipeline.contract_pipeline_name;
 26 pub const contract_pipeline_description = pipeline.contract_pipeline_description;
 27 pub const tensor_pipeline_name = pipeline.tensor_pipeline_name;
 28 pub const tensor_pipeline_description = pipeline.tensor_pipeline_description;
 29 pub const dispatch_pipeline_name = pipeline.dispatch_pipeline_name;
 30 pub const dispatch_pipeline_description = pipeline.dispatch_pipeline_description;
 31 pub const memory_pipeline_name = pipeline.memory_pipeline_name;
 32 pub const memory_pipeline_description = pipeline.memory_pipeline_description;
 33 pub const kernel_pipeline_name = pipeline.kernel_pipeline_name;
 34 pub const kernel_pipeline_description = pipeline.kernel_pipeline_description;
 35 pub const target_pipeline_name = pipeline.target_pipeline_name;
 36 pub const target_pipeline_description = pipeline.target_pipeline_description;
 37 pub const BackendTargetProfile = pipeline.BackendTargetProfile;
 38 pub const canonicalizeModule = pipeline.canonicalizeModule;
 39 pub const saturateTensorOperations = pipeline.saturateTensorOperations;
 40 pub const setBackendTargetProfile = pipeline.setBackendTargetProfile;
 41 pub const readBackendTargetProfile = pipeline.readBackendTargetProfile;
 42 pub const backend_legalization_pass_name = pipeline.backend_legalization_pass_name;
 43 pub const bufferization_planning_pass_name = pipeline.bufferization_planning_pass_name;
 44 pub const canonicalization_pass_name = pipeline.canonicalization_pass_name;
 45 pub const constant_folding_pass_name = pipeline.constant_folding_pass_name;
 46 pub const saturation_pass_name = pipeline.saturation_pass_name;
 47 pub const dtype_legalization_pass_name = pipeline.dtype_legalization_pass_name;
 48 pub const activation_lowering_pass_name = pipeline.activation_lowering_pass_name;
 49 pub const einsum_lowering_pass_name = pipeline.einsum_lowering_pass_name;
 50 pub const indexing_lowering_pass_name = pipeline.indexing_lowering_pass_name;
 51 pub const fusion_planning_pass_name = pipeline.fusion_planning_pass_name;
 52 pub const kernelization_pass_name = pipeline.kernelization_pass_name;
 53 pub const kernel_outlining_planning_pass_name = pipeline.kernel_outlining_planning_pass_name;
 54 pub const layout_planning_pass_name = pipeline.layout_planning_pass_name;
 55 pub const memory_space_planning_pass_name = pipeline.memory_space_planning_pass_name;
 56 pub const schedule_planning_pass_name = pipeline.schedule_planning_pass_name;
 57 pub const shape_layout_pass_name = pipeline.shape_layout_pass_name;
 58 pub const PipelineError = pipeline.PipelineError;
 59 pub const BackendPreparationStats = pipeline.BackendPreparationStats;
 60 pub const BackendPreparationFailureKind = pipeline.BackendPreparationFailureKind;
 61 pub const BackendPreparationFailure = pipeline.BackendPreparationFailure;
 62 pub const BackendPreparationTiming = pipeline.BackendPreparationTiming;
 63 pub const BackendPreparationRunOptions = pipeline.BackendPreparationRunOptions;
 64 pub const BackendPreparationRun = pipeline.BackendPreparationRun;
 65 pub const BackendPreparationProductStamps = pipeline.BackendPreparationProductStamps;
 66 pub const BackendPreparationProductKeys = pipeline.BackendPreparationProductKeys;
 67 pub const BackendPreparedJob = pipeline.BackendPreparedJob;
 68 pub const ContractPreparationResult = pipeline.ContractPreparationResult;
 69 pub const TensorPreparationResult = pipeline.TensorPreparationResult;
 70 pub const DispatchPreparationResult = pipeline.DispatchPreparationResult;
 71 pub const MemoryPreparationResult = pipeline.MemoryPreparationResult;
 72 pub const KernelPreparationResult = pipeline.KernelPreparationResult;
 73 pub const TargetPreparationResult = pipeline.TargetPreparationResult;
 74 pub const BackendPreparationCache = cache.BackendPreparationCache;
 75 pub const BackendPreparationCacheUpdate = cache.BackendPreparationCacheUpdate;
 76 pub const KernelizationProduct = kernelization.KernelizationAnalysis;
 77 pub const GeneratedScheduleKind = kernelization.GeneratedScheduleKind;
 78 pub const GeneratedSchedule = kernelization.GeneratedSchedule;
 79 pub const GeneratedKernelProgram = kernelization.GeneratedKernelProgram;
 80 pub const GeneratedKernelSummary = kernelization.GeneratedKernelSummary;
 81 pub const GeneratedKernelSummaries = kernelization.GeneratedKernelSummaries;
 82 pub const KernelLibraryLowering = pipeline.KernelLibraryLowering;
 83 pub const ActivationLoweringOptions = pipeline.ActivationLoweringOptions;
 84 pub const EinsumLoweringOptions = pipeline.EinsumLoweringOptions;
 85 pub const IndexingLoweringOptions = pipeline.IndexingLoweringOptions;
 86 pub const TensorLoweringOptions = pipeline.TensorLoweringOptions;
 87 pub const runTargetPipelineWithDiagnostics = pipeline.runTargetPipelineWithDiagnostics;
 88 pub const CatalogCallOptions = call.CatalogCallOptions;
 89 pub const CatalogCallRequest = call.CatalogCallRequest;
 90 pub const backendLegalizationPass = pipeline.backendLegalizationPass;
 91 pub const bufferizationPlanningPass = pipeline.bufferizationPlanningPass;
 92 pub const canonicalizationPass = pipeline.canonicalizationPass;
 93 pub const canonicalization_pattern_entries = canonicalization.canonicalization_pattern_entries;
 94 pub const shapeLayoutPropagationPass = pipeline.shapeLayoutPropagationPass;
 95 pub const constantFoldingPass = pipeline.constantFoldingPass;
 96 pub const tensorSaturationPass = pipeline.tensorSaturationPass;
 97 pub const dtypeLegalizationPass = pipeline.dtypeLegalizationPass;
 98 pub const activationLoweringPass = pipeline.activationLoweringPass;
 99 pub const activationLoweringPassWithOptions = pipeline.activationLoweringPassWithOptions;
100 pub const einsumLoweringPass = pipeline.einsumLoweringPass;
101 pub const einsumLoweringPassWithOptions = pipeline.einsumLoweringPassWithOptions;
102 pub const indexingLoweringPass = pipeline.indexingLoweringPass;
103 pub const indexingLoweringPassWithOptions = pipeline.indexingLoweringPassWithOptions;
104 pub const lossLoweringPass = pipeline.lossLoweringPass;
105 pub const lossLoweringPassWithOptions = pipeline.lossLoweringPassWithOptions;
106 pub const fusionPlanningPass = pipeline.fusionPlanningPass;
107 pub const schedulePlanningPass = pipeline.schedulePlanningPass;
108 pub const kernelOutliningPlanningPass = pipeline.kernelOutliningPlanningPass;
109 pub const kernelizationPass = pipeline.kernelizationPass;
110 pub const memorySpacePlanningPass = pipeline.memorySpacePlanningPass;
111 pub const layoutPlanningPass = pipeline.layoutPlanningPass;
112 pub const BackendPreparationStage = pipeline.BackendPreparationStage;
113 pub const contract_stages = pipeline.contract_stages;
114 pub const tensor_stages = pipeline.tensor_stages;
115 pub const dispatch_stages = pipeline.dispatch_stages;
116 pub const memory_stages = pipeline.memory_stages;
117 pub const kernel_stages = pipeline.kernel_stages;
118 pub const target_stages = pipeline.target_stages;
119 pub const contract_pipeline_registration = pipeline.contract_pipeline_registration;
120 pub const tensor_pipeline_registration = pipeline.tensor_pipeline_registration;
121 pub const dispatch_pipeline_registration = pipeline.dispatch_pipeline_registration;
122 pub const memory_pipeline_registration = pipeline.memory_pipeline_registration;
123 pub const kernel_pipeline_registration = pipeline.kernel_pipeline_registration;
124 pub const target_pipeline_registration = pipeline.target_pipeline_registration;
125 pub const contract_pass_registrations = pipeline.contract_pass_registrations;
126 pub const tensor_pass_registrations = pipeline.tensor_pass_registrations;
127 pub const dispatch_pass_registrations = pipeline.dispatch_pass_registrations;
128 pub const memory_pass_registrations = pipeline.memory_pass_registrations;
129 pub const kernel_pass_registrations = pipeline.kernel_pass_registrations;
130 pub const target_pass_registrations = pipeline.target_pass_registrations;
131 pub const accy_choir_pass_registrations = pipeline.accy_choir_pass_registrations;
132 pub const accy_choir_package_extension = pipeline.accy_choir_package_extension;
133 pub const contract_pass_plan = pipeline.contract_pass_plan;
134 pub const tensor_pass_plan = pipeline.tensor_pass_plan;
135 pub const dispatch_pass_plan = pipeline.dispatch_pass_plan;
136 pub const memory_pass_plan = pipeline.memory_pass_plan;
137 pub const kernel_pass_plan = pipeline.kernel_pass_plan;
138 pub const target_pass_plan = pipeline.target_pass_plan;
139 pub const contract_pass_count = pipeline.contract_pass_count;
140 pub const contract_analysis_count = pipeline.contract_analysis_count;
141 pub const tensor_pass_count = pipeline.tensor_pass_count;
142 pub const tensor_analysis_count = pipeline.tensor_analysis_count;
143 pub const dispatch_pass_count = pipeline.dispatch_pass_count;
144 pub const dispatch_analysis_count = pipeline.dispatch_analysis_count;
145 pub const memory_pass_count = pipeline.memory_pass_count;
146 pub const memory_analysis_count = pipeline.memory_analysis_count;
147 pub const kernel_pass_count = pipeline.kernel_pass_count;
148 pub const kernel_analysis_count = pipeline.kernel_analysis_count;
149 pub const target_pass_count = pipeline.target_pass_count;
150 pub const target_analysis_count = pipeline.target_analysis_count;
151 pub const contractPassName = pipeline.contractPassName;
152 pub const contractAnalysisName = pipeline.contractAnalysisName;
153 pub const tensorPassName = pipeline.tensorPassName;
154 pub const tensorAnalysisName = pipeline.tensorAnalysisName;
155 pub const dispatchPassName = pipeline.dispatchPassName;
156 pub const dispatchAnalysisName = pipeline.dispatchAnalysisName;
157 pub const memoryPassName = pipeline.memoryPassName;
158 pub const memoryAnalysisName = pipeline.memoryAnalysisName;
159 pub const kernelPassName = pipeline.kernelPassName;
160 pub const kernelAnalysisName = pipeline.kernelAnalysisName;
161 pub const targetPassName = pipeline.targetPassName;
162 pub const targetAnalysisName = pipeline.targetAnalysisName;
163 pub const addContractPipeline = pipeline.addContractPipeline;
164 pub const addTensorPipeline = pipeline.addTensorPipeline;
165 pub const addTensorPipelineWithOptions = pipeline.addTensorPipelineWithOptions;
166 pub const addDispatchPipeline = pipeline.addDispatchPipeline;
167 pub const addMemoryPipeline = pipeline.addMemoryPipeline;
168 pub const addKernelPipeline = pipeline.addKernelPipeline;
169 pub const addTargetPipeline = pipeline.addTargetPipeline;
170 pub const runTargetPipeline = pipeline.runTargetPipeline;
171 pub const runBackendPreparationPipelineFromSemanticModule = pipeline.runBackendPreparationPipelineFromSemanticModule;
172 pub const prepareContractJobFromSemanticModule = pipeline.prepareContractJobFromSemanticModule;
173 pub const prepareTensorJobFromContractJob = pipeline.prepareTensorJobFromContractJob;
174 pub const prepareDispatchJobFromTensorJob = pipeline.prepareDispatchJobFromTensorJob;
175 pub const prepareMemoryJobFromDispatchJob = pipeline.prepareMemoryJobFromDispatchJob;
176 pub const prepareKernelJobFromMemoryJob = pipeline.prepareKernelJobFromMemoryJob;
177 pub const prepareTargetJobFromKernelJob = pipeline.prepareTargetJobFromKernelJob;
178 pub const prepareBackendJobFromSemanticModule = pipeline.prepareBackendJobFromSemanticModule;
179 pub const prepareBackendJobFromContractJob = pipeline.prepareBackendJobFromContractJob;
180 pub const prepareContractJobFromSemanticModuleWithRun = pipeline.prepareContractJobFromSemanticModuleWithRun;
181 pub const prepareTensorJobFromContractJobWithRun = pipeline.prepareTensorJobFromContractJobWithRun;
182 pub const prepareDispatchJobFromTensorJobWithRun = pipeline.prepareDispatchJobFromTensorJobWithRun;
183 pub const prepareMemoryJobFromDispatchJobWithRun = pipeline.prepareMemoryJobFromDispatchJobWithRun;
184 pub const prepareKernelJobFromMemoryJobWithRun = pipeline.prepareKernelJobFromMemoryJobWithRun;
185 pub const prepareTargetJobFromKernelJobWithRun = pipeline.prepareTargetJobFromKernelJobWithRun;
186 pub const prepareBackendJobFromContractPreparationResult = pipeline.prepareBackendJobFromContractPreparationResult;
187 pub const prepareBackendJobFromTensorPreparationResult = pipeline.prepareBackendJobFromTensorPreparationResult;
188 pub const prepareBackendJobFromDispatchPreparationResult = pipeline.prepareBackendJobFromDispatchPreparationResult;
189 pub const prepareBackendJobFromMemoryPreparationResult = pipeline.prepareBackendJobFromMemoryPreparationResult;
190 pub const prepareBackendJobFromKernelPreparationResult = pipeline.prepareBackendJobFromKernelPreparationResult;
191 pub const prepareBackendJobFromTargetPreparationResult = pipeline.prepareBackendJobFromTargetPreparationResult;
192 pub const buildBackendPreparationContext = pipeline.buildBackendPreparationContext;
193 pub const insertCatalogCall = call.insertCatalogCall;