lib/accy/src/kernel/root.zig

daab053ee43316e1809a84551d573ddd1e5bf3d2

  1 const gpu = @import("gpu");
  2 const model = @import("model/root.zig");
  3 pub const builder = model.core.builder;
  4 const call = @import("call.zig");
  5 pub const domain = model.core.domain;
  6 const dsl = @import("dsl/root.zig");
  7 const compile = @import("compile/root.zig");
  8 const interpret_mod = @import("interpret/root.zig");
  9 const library_mod = @import("library/root.zig");
 10 const limits = model.core;
 11 const logical_mod = @import("logical/root.zig");
 12 pub const oracle = @import("oracle/root.zig");
 13 pub const plan = model.plan;
 14 pub const program = @import("program/root.zig");
 15 pub const schedule = model.core.schedule;
 16 pub const typed = model.core.typed;
 17 pub const vector = model.core.vector;
 18 pub const view = model.core.view;
 19 pub const artifact = compile.artifact;
 20 
 21 pub const Param = builder.Param;
 22 pub const Buffer = builder.Buffer;
 23 pub const Type = builder.Type;
 24 pub const Kernel = builder.Kernel;
 25 pub const RawBuilder = builder.Builder;
 26 pub const RawStorage = builder.Storage;
 27 pub const Builder = program.Builder;
 28 pub const MmaShape = builder.MmaShape;
 29 pub const Value = builder.Value;
 30 pub const Block = builder.Block;
 31 pub const If = builder.If;
 32 pub const For = builder.For;
 33 pub const ForScope = builder.ForScope;
 34 pub const While = builder.While;
 35 pub const WhileScope = builder.WhileScope;
 36 pub const InsertionScope = builder.InsertionScope;
 37 pub const VerificationDiagnostic = builder.VerificationDiagnostic;
 38 pub const Plan = plan.Plan;
 39 pub const PlanOptions = plan.Options;
 40 pub const PlanError = plan.Error;
 41 pub const PlanVersion = plan.plan_version;
 42 pub const BackendHandle = gpu.BackendHandle;
 43 pub const KernelArtifact = gpu.KernelArtifact;
 44 pub const KernelCallArtifactOptions = call.Options;
 45 pub const OwnedKernelCallArtifact = call.OwnedArtifact;
 46 pub const Argument = oracle.Argument;
 47 pub const Scalar = oracle.Scalar;
 48 pub const ExecutionDiagnostic = oracle.ExecutionDiagnostic;
 49 pub const Graph = program.Program;
 50 pub const Program = dsl.Program;
 51 pub const Family = dsl.Family;
 52 pub const program_product_name = program.product_name;
 53 pub const interpret = interpret_mod;
 54 pub const library = library_mod;
 55 pub const logical = logical_mod;
 56 pub const DomainAxis = domain.DomainAxis;
 57 pub const Domain2D = domain.Domain2D;
 58 pub const Domain3D = domain.Domain3D;
 59 pub const Index1D = domain.Index1D;
 60 pub const VectorIndex1D = domain.VectorIndex1D;
 61 pub const Index2D = domain.Index2D;
 62 pub const Index3D = domain.Index3D;
 63 pub const BufferView = view.BufferView;
 64 pub const KeyBufferView = view.KeyBufferView;
 65 pub const TypedValue = typed.Value;
 66 pub const KeyValue = typed.KeyValue;
 67 pub const TypedVec2 = typed.Vec2;
 68 pub const TypedVec3 = typed.Vec3;
 69 pub const Vec2 = vector.Vec2;
 70 pub const Vec3 = vector.Vec3;
 71 pub const Guard = program.Guard;
 72 pub const Schedule = schedule.Schedule;
 73 pub const ScheduleError = schedule.ScheduleError;
 74 pub const Launch = schedule.Launch;
 75 pub const Limits = limits.Limits;
 76 pub const Capacity = program.Builder.Capacity;
 77 pub const RawLimits = limits.RawLimits;
 78 pub const RawCapacity = builder.Builder.Capacity;
 79 pub const RawUsage = builder.Storage.Usage;
 80 pub const RawExhaustion = builder.Storage.Exhaustion;
 81 pub const ScheduleLimits = schedule.ScheduleLimits;
 82 pub const ScheduleCapacity = schedule.ScheduleCapacity;
 83 pub const ScheduleUsage = schedule.ScheduleUsage;
 84 pub const ScheduleExhaustion = schedule.ScheduleExhaustion;
 85 pub const ScheduleSnapshotLimits = schedule.SnapshotLimits;
 86 pub const ScheduleSnapshotCapacity = schedule.SnapshotCapacity;
 87 pub const ScheduleSnapshotUsage = schedule.SnapshotUsage;
 88 pub const ScheduleSnapshotExhaustion = schedule.SnapshotExhaustion;
 89 pub const Axis = schedule.Axis;
 90 pub const AxisId = schedule.AxisId;
 91 pub const AxisStep = schedule.AxisStep;
 92 pub const BindTarget = schedule.BindTarget;
 93 pub const ScheduleSnapshot = schedule.Snapshot;
 94 pub const ScheduleSnapshotVersion = schedule.snapshot_version;
 95 pub const Split = schedule.Split;
 96 pub const Step = schedule.Step;
 97 pub const scalar = builder.scalar;
 98 pub const buffer = builder.buffer;
 99 pub const dynamicBuffer = builder.dynamicBuffer;
100 pub const domainAxis = domain.domainAxis;
101 pub const createPlan = plan.create;
102 pub const createArtifactJob = artifact.createJob;
103 pub const createArtifactPlan = artifact.createPlan;
104 pub const createKernelCallArtifact = call.createArtifact;
105 pub const createBackendArtifactFromKernelCallEntry = call.createBackendArtifactFromEntry;
106 pub const kernelCallEntryLaunchGeometry = call.launchGeometryForEntry;
107 pub const compileFragment = compile.compileFragment;
108 pub const createKernelArtifact = compile.createArtifact;
109 pub const argumentBuffer = oracle.argumentBuffer;
110 pub const argumentBool = oracle.argumentBool;
111 pub const argumentI32 = oracle.argumentI32;
112 pub const argumentU32 = oracle.argumentU32;
113 pub const argumentI64 = oracle.argumentI64;
114 pub const argumentF32 = oracle.argumentF32;
115 pub const argumentF64 = oracle.argumentF64;
116 pub const Compare = builder.Compare;
117 pub const Dimension = builder.Dimension;
118 pub const AddressSpace = builder.AddressSpace;
119 pub const MemoryOrder = builder.MemoryOrder;
120 pub const Scope = builder.Scope;
121 pub const WarpOpKind = builder.WarpOpKind;