Home
last modified time | relevance | path

Searched full:batch (Results 1 – 25 of 163) sorted by relevance

1234567

/llvm-project/mlir/unittests/Dialect/Utils/
H A DStructuredOpsUtilsTest.cpp194 AffineExpr batch, m, n, k; in TEST() local
195 bindDims(&context, batch, m, n, k); in TEST()
196 auto mapA = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, k}, &context)); in TEST()
197 auto mapB = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, n}, &context)); in TEST()
198 auto mapC = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, n}, &context)); in TEST()
207 AffineExpr batch, m, n, k; in TEST() local
208 bindDims(&context, k, batch, m, n); // bind in different order in TEST()
209 auto mapA = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, k}, &context)); in TEST()
210 auto mapB = AffineMapAttr::get(AffineMap::get(4, 0, {batch, k, n}, &context)); in TEST()
211 auto mapC = AffineMapAttr::get(AffineMap::get(4, 0, {batch, m, n}, &context)); in TEST()
[all …]
/llvm-project/compiler-rt/lib/scudo/standalone/
H A Dallocator_common.h25 memcpy(Batch, Array, sizeof(Batch[0]) * Count); in setFromArray()
29 memcpy(Batch + Count, Array, sizeof(Batch[0]) * N); in appendFromArray()
37 memcpy(Batch + Count, B->Batch + (B->Count - N), sizeof(Batch[0]) * N); in appendFromTransferBatch()
46 Batch[Count++] = P; in add()
49 memcpy(Array, Batch, sizeof(Batch[ in moveToArray()
[all...]
H A Dquarantine.h20 // With the following count, a batch (and the header that protects it) occupy
26 void *Batch[MaxCount]; member
30 Batch[0] = Ptr; in init()
31 this->Size = Size + sizeof(QuarantineBatch); // Account for the Batch Size. in init()
34 // The total size of quarantined nodes recorded in this batch.
39 Batch[Count++] = Ptr; in push_back()
52 Batch[Count + I] = From->Batch[I]; in merge()
60 void shuffle(u32 State) { ::scudo::shuffle(Batch, Count, &State); } in shuffle()
114 // Move all the chunks into the current batch. in mergeBatches()
118 // Remove the next batch From the list and account for its Size. in mergeBatches()
[all …]
/llvm-project/llvm/lib/ExecutionEngine/Orc/Debugging/
H A DVTuneSupportPlugin.cpp27 VTuneMethodBatch Batch; in getMethodBatch() local
41 &Batch](StringRef S) mutable { in getMethodBatch()
44 Batch.Strings.push_back(S.str()); in getMethodBatch()
45 I->second = Batch.Strings.size(); in getMethodBatch()
53 Batch.Methods.push_back(VTuneMethodInfo()); in getMethodBatch()
54 auto &Method = Batch.Methods.back(); in getMethodBatch()
73 Method.SourceFileSI = Batch.Strings.size(); in getMethodBatch()
74 Batch.Strings.push_back(DC->getLineInfoForAddress(SAddr).FileName); in getMethodBatch()
81 return Batch; in getMethodBatch()
89 auto Batch in modifyPassConfig() local
[all...]
H A DPerfSupportPlugin.cpp229 PerfJITRecordBatch Batch; in getRecords()
236 Batch.DebugInfoRecords.push_back(std::move(*DebugInfo)); in getRecords()
238 Batch.CodeLoadRecords.push_back(getCodeLoadRecord(*Sym, CodeIndex)); in getRecords()
245 Batch.UnwindingRecord = std::move(*UWR); in getRecords()
248 Batch.UnwindingRecord.Prefix.TotalSize = 0; in getRecords()
250 return Batch; in getRecords()
273 auto Batch = getRecords(EPC.getExecutionSession(), G, CodeIndex, in modifyPassConfig()
278 RegisterPerfImplAddr, Batch)), in modifyPassConfig()
230 PerfJITRecordBatch Batch; getRecords() local
274 auto Batch = getRecords(EPC.getExecutionSession(), G, CodeIndex, modifyPassConfig() local
/llvm-project/compiler-rt/lib/sanitizer_common/
H A Dsanitizer_quarantine.h32 void *batch[kSize]; member
36 batch[0] = ptr; in init()
37 this->size = size + sizeof(QuarantineBatch); // Account for the batch size. in init()
40 // The total size of quarantined nodes recorded in this batch.
47 batch[count++] = ptr; in push_back()
60 batch[count + i] = from->batch[i]; in merge()
161 // require some tuning). It saves us merge attempt when the batch list in Recycle()
182 CHECK(kPrefetch <= ARRAY_SIZE(b->batch)); in DoRecycle()
184 PREFETCH(b->batch[i]); in DoRecycle()
187 PREFETCH(b->batch[i + kPrefetch]); in DoRecycle()
[all …]
H A Dsanitizer_allocator_local_cache.h168 void *res = c->batch[--c->count]; in Allocate()
169 PREFETCH(c->batch[c->count - 1]); in Allocate()
183 c->batch[c->count++] = p; in Deallocate()
211 void *batch[2 * TransferBatch::kMaxNumCached]; member
227 // id. 0 means the class size is large enough to store a batch within one in InitCache()
248 b->CopyToArray(c->batch); in Refill()
259 class_id, allocator, (TransferBatch *)c->batch[first_idx_to_drain]); in Drain()
260 // Failure to allocate a batch while releasing memory is non recoverable. in Drain()
261 // TODO(alekseys): Figure out how to do it without allocating a new batch. in Drain()
264 "transfer batch.\n", SanitizerToolName); in Drain()
[all …]
/llvm-project/compiler-rt/lib/tsan/rtl/
H A Dtsan_dense_alloc.h113 // The stack itself uses Block::next links, while the batch within each
114 // stack node uses Block::batch links.
122 IndexT batch; member
131 // Pop 1 batch of nodes from the freelist. in Refill()
146 idx = MapBlock(idx)->batch; in Refill()
151 // Build a batch of at most Cache::kSize / 2 nodes linked by Block::batch. in Drain()
156 ptr->batch = head_idx; in Drain()
180 T *batch = (T *)MmapOrDie(kL2Size * sizeof(T), name_); in AllocSuperBlock() local
181 map_[fillpos] = batch; in AllocSuperBlock()
184 new (batch + i) T; in AllocSuperBlock()
/llvm-project/mlir/test/Integration/Dialect/Transform/
H A Dmatch_batch_matmul.mlir16 %fill, %bmm, %dims, %lhs_type, %rhs_type, %res_type, %batch, %m, %n, %k =
23 transform.yield %fill, %bmm, %dims, %lhs_type, %rhs_type, %res_type, %batch
34 %batch: !transform.param<i64> {transform.readonly}) {
36 transform.debug.emit_remark_at %bmm, "batch matmul" : !transform.any_op
41 …transform.debug.emit_param_as_remark %batch, "batch dimension" at %bmm : !transform.param<i64>, !t…
58 // expected-remark @below {{batch matmul}}
63 // expected-remark @below {{batch dimension 0}}
/llvm-project/lldb/test/API/driver/batch_mode/
H A DTestBatchMode.py2 Test that the lldb driver's batch mode works correctly.
20 """Test that the lldb driver's batch mode works correctly."""
25 # Pass CRASH so the process will crash and stop in batch mode.
58 """Test that the lldb driver's batch mode works correctly."""
95 """Test that the lldb driver's batch mode works correctly for process launch."""
102 # this is not a reason to exit batch mode.
136 """Test that the lldb driver's batch mode works correctly."""
144 # care must be taken not to treat that as a reason to exit batch mode.
/llvm-project/mlir/include/mlir/Dialect/SparseTensor/IR/
H A DSparseTensorStorageLayout.h33 /// memref<[batch] x ? x pos> positions ; positions for level l
34 /// memref<[batch] x ? x crd> coordinates ; coordinates for level l
35 /// ; if loose-[batch] x compressed:
36 /// memref<[batch] x ? x pos> positions ; lo/hi pos pairs for level l
37 /// memref<[batch] x ? x crd> coordinates ; coordinates for level l
39 /// memref<[batch] x ? x crd> coordinates ; coordinates for level l
41 /// memref<[batch] x ? x eltType> values ; values
45 /// // TODO: memSizes need to be expanded to array<[batch] x n x int> to
47 /// // assume that every batch occupies the same memory size.
H A DEnums.h157 Batch = 0x00020000, enumerator
171 encPowOfTwo(LevelFormat::Batch) &&
189 case LevelFormat::Batch: in toFormatString()
190 return "batch"; in toFormatString()
244 // If undefined/dense/batch/NOutOfM, then must be unique and ordered. in isValidLvlBits()
247 LevelFormat::Batch, LevelFormat::NOutOfM>(fmt)) in isValidLvlBits()
344 return isa<LevelFormat::Dense, LevelFormat::Batch>(); in hasDenseSemantic()
414 inline bool isBatchLT(LevelType lt) { return lt.isa<LevelFormat::Batch>(); } in isBatchLT()
/llvm-project/compiler-rt/lib/sanitizer_common/tests/
H A Dsanitizer_quarantine_test.cpp38 while (QuarantineBatch *batch = cache->DequeueBatch()) in DeallocateCache() local
39 cb.Deallocate(batch); in DeallocateCache()
52 ASSERT_EQ(into.batch[0], kFakePtr); in TEST()
53 ASSERT_EQ(into.batch[1], kFakePtr); in TEST()
61 // Merge the batch to the limit. in TEST()
115 // Batches merged, one batch to deallocate. in TEST()
/llvm-project/compiler-rt/lib/scudo/standalone/tests/
H A Drelease_test.cpp165 Batch[Count++] = P; in add()
170 return Batch[I]; in get()
175 scudo::uptr Batch[MaxCount]; member
180 typedef FreeBatch<SizeClassMap> Batch; in testReleaseFreeMemoryToOS() typedef
209 scudo::SinglyLinkedList<Batch> FreeList; in testReleaseFreeMemoryToOS()
211 Batch *CurrentBatch = nullptr; in testReleaseFreeMemoryToOS()
214 CurrentBatch = new Batch; in testReleaseFreeMemoryToOS()
219 if (CurrentBatch->getCount() == Batch::MaxCount) in testReleaseFreeMemoryToOS()
416 typedef FreeBatch<SizeClassMap> Batch; in testReleasePartialRegion() typedef
433 scudo::SinglyLinkedList<Batch> FreeList; in testReleasePartialRegion()
[all …]
H A Dquarantine_test.cpp32 while (scudo::QuarantineBatch *Batch = Cache->dequeueBatch()) in deallocateCache() local
33 Cb.deallocate(Batch); in deallocateCache()
46 EXPECT_EQ(Into.Batch[0], FakePtr); in TEST()
47 EXPECT_EQ(Into.Batch[1], FakePtr); in TEST()
55 // Merge the batch to the limit. in TEST()
117 // Batches merged, one batch to deallocate. in TEST()
/llvm-project/libc/benchmarks/
H A DLibcBenchmark.h49 Last, // Keep only the last batch.
169 // The batch size is growing by `ScalingFactor` to minimize the effect of
181 // batch to mitigate the effect of the branch predictor. Signature:
208 // Request a new Batch of size `Iterations`.
209 const auto &Batch = PP.generateBatch(Iterations);
211 // Measuring this Batch.
213 for (const auto Parameter : Batch) {
/llvm-project/lldb/tools/driver/
H A DOptions.td108 def batch: F<"batch">,
112 Alias<batch>,
113 HelpText<"Alias for --batch">,
126 …HelpText<"When in batch mode, tells the debugger to run this one-line lldb command if the target c…
135 …HelpText<"When in batch mode, tells the debugger to source this file of lldb commands if the targe…
/llvm-project/mlir/include/mlir/Dialect/Linalg/TransformOps/
H A DLinalgMatchOps.td148 C(batch, m, n) += A(batch, m, k) * B(batch, k, n)
152 - 'batch' are parallel dimensions used in inputs and result;
168 let results = (outs TransformParamTypeInterface:$batch,
189 C(batch, depth, oi, oc) += A(batch, depth, oi, ic) * B(fl, depth, ic, oc)
193 - 'batch' are parallel dimensions used in the input and result;
215 let results = (outs TransformParamTypeInterface:$batch,
/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgInterfaces.h44 SmallVector<unsigned, 2> batch; member
59 /// 5. Optional batch dimensions that appear in all operands are captured.
62 /// When multiple dimension occurrences exist that match `batch`, `m`, `n`, or
76 SmallVector<unsigned, 2> batch; member
91 /// 1. Optional batch dimensions that appear in the input and filter.
/llvm-project/mlir/include/mlir/Dialect/Tosa/Utils/
H A DConversionUtils.h47 // Checks for a dynamic batch dim in any of the passed parameters of an op.
48 // The batch dimention must be #0 and the rest of the dimensions must be static.
67 op, "input can only be dynamic for batch size"); in checkHasDynamicBatchDims()
/llvm-project/llvm/lib/ExecutionEngine/Orc/TargetProcess/
H A DJITLoaderPerf.cpp183 static Error registerJITLoaderPerfImpl(const PerfJITRecordBatch &Batch) {
188 // Serialize the batch in registerJITLoaderPerfImpl()
190 if (Batch.UnwindingRecord.Prefix.TotalSize > 0) in registerJITLoaderPerfImpl()
191 writeUnwindRecord(Batch.UnwindingRecord); in registerJITLoaderPerfImpl()
193 for (const auto &DebugInfo : Batch.DebugInfoRecords) in registerJITLoaderPerfImpl()
196 for (const auto &CodeLoad : Batch.CodeLoadRecords) in registerJITLoaderPerfImpl()
434 static Error badOSBatch(PerfJITRecordBatch &Batch) { return badOS(); }
184 registerJITLoaderPerfImpl(const PerfJITRecordBatch & Batch) registerJITLoaderPerfImpl() argument
435 badOSBatch(PerfJITRecordBatch & Batch) badOSBatch() argument
/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py6 Batch = S.Batch variable
468 lhs=TensorDef(TV.LhsType, Batch, S.M, S.K, S.M0, S.K0),
469 rhs=TensorDef(TV.RhsType, Batch, S.N, S.K, S.N0, S.K0),
470 accum=TensorDef(TV.AccumType, Batch, S.M, S.N, S.M0, S.N0, output=True),
475 Besides the outermost batch dimension has the same semantic as
477 non-batch dimensions are the same as linalg.mmt4d vs. linalg.matmul. See the
489 A=TensorDef(T1, Batch, S.M, S.K),
490 B=TensorDef(T2, Batch, S.K, S.N),
491 C=TensorDef(U, Batch, argument
512 batch_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument
530 batch_matmul_transpose_a( A=TensorDef(T1, Batch, S.K, S.M), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument
549 batch_matmul_transpose_b( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.N, S.K), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument
568 quantized_batch_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), AZp=ScalarDef(I32), BZp=ScalarDef(I32), C=TensorDef(U, Batch, S.M, S.N, output=True), ) global() argument
589 batch_reduce_matmul( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, S.M, S.N, output=True), ) global() argument
636 batch_matvec( A=TensorDef(T1, Batch, S.M, S.K), B=TensorDef(T2, Batch, S.K), C=TensorDef(U, Batch, S.M, output=True), ) global() argument
654 batch_vecmat( A=TensorDef(T1, Batch, S.K), B=TensorDef(T2, Batch, S.K, S.N), C=TensorDef(U, Batch, S.N, output=True), ) global() argument
[all...]
/llvm-project/lldb/docs/man/
H A Dlldb.rst56 .. option:: --batch
62 Alias for --batch
86 When in batch mode, tells the debugger to run this one-line lldb command if the target crashes.
110 When in batch mode, tells the debugger to source this file of lldb commands if the target crashes.
/llvm-project/mlir/test/Examples/transform/Ch4/
H A Dfeatures.mlir99 %batch, %lhs, %rhs, %reduction =
105 // There is one of lhs, rhs and reduction dimensions and zero batch
107 %n_batch = transform.num_associations %batch
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp402 /// 5. Optional batch dimensions that appear in all operands are captured. in isContractionInterfaceImpl()
423 // A & B & C are the "batch" dimensions. in isContractionInterfaceImpl()
441 llvm::sort(dimensions.batch.begin(), dimensions.batch.end()); in getMatchContractionMessage()
718 // unConvolvedDims & outputDims - filterDims are the batch iterators.
719 llvm::SmallDenseSet<int64_t> batch = inputExprWalker.unConvolvedDims;
720 llvm::set_intersect(batch, outputDims);
721 llvm::set_subtract(batch, filterDims);
754 SmallVector<unsigned, 2>(batch.begin(), batch in inferConvolutionDims()
641 llvm::SmallDenseSet<int64_t> batch = inputExprWalker.unConvolvedDims; inferConvolutionDimsImpl() local
[all...]

1234567