Home
last modified time | relevance | path

Searched refs:memory_space (Results 1 – 12 of 12) sorted by relevance

/llvm-project-15.0.7/mlir/test/Dialect/Bufferization/Transforms/
H A Dtensor-copy-insertion-memory-space.mlir20 …// CHECK: bufferization.alloc_tensor() {bufferization.escape = [false], memory_space = 1 : ui64} :…
21 %t = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<10xf32>
22 …// CHECK: bufferization.alloc_tensor() {bufferization.escape = [false], memory_space = 1 : ui64} :…
H A Dtensor-copy-insertion.mlir43 …// CHECK: bufferization.alloc_tensor() {bufferization.escape = [false], memory_space = 0 : ui64} :…
46 …// CHECK-NO-DEALLOC: bufferization.alloc_tensor() {bufferization.escape = [true], memory_space = 0…
58 …:.*]] = bufferization.alloc_tensor() {bufferization.escape = [false], memory_space = 0 : ui64} : t…
77 …:.*]] = bufferization.alloc_tensor() {bufferization.escape = [false], memory_space = 0 : ui64} : t…
H A Done-shot-bufferize.mlir146 %0 = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<5xf32>
/llvm-project-15.0.7/mlir/test/Dialect/SCF/
H A Done-shot-bufferize-invalid.mlir6 %0 = bufferization.alloc_tensor() {memory_space = 0 : ui64} : tensor<10xf32>
7 %1 = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<10xf32>
H A Done-shot-bufferize.mlir643 %0 = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<5xf32>
673 %1 = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<5xf32>
693 %A = bufferization.alloc_tensor(%sz) {memory_space = 1 : ui64} : tensor<?xf32>
694 %B = bufferization.alloc_tensor(%sz) {memory_space = 1 : ui64} : tensor<?xf32>
/llvm-project-15.0.7/mlir/test/Dialect/Arithmetic/
H A Done-shot-bufferize-memory-space-invalid.mlir6 %0 = bufferization.alloc_tensor() {memory_space = 0 : ui64} : tensor<10xf32>
7 %1 = bufferization.alloc_tensor() {memory_space = 1 : ui64} : tensor<10xf32>
/llvm-project-15.0.7/polly/lib/External/ppcg/
H A Dgpu_print.c83 const char *memory_space) in gpu_array_info_print_declaration_argument() argument
92 if (memory_space) { in gpu_array_info_print_declaration_argument()
93 p = isl_printer_print_str(p, memory_space); in gpu_array_info_print_declaration_argument()
H A Dgpu_print.h19 const char *memory_space);
/llvm-project-15.0.7/mlir/test/python/ir/
H A Dbuiltin_types.py371 memref = MemRefType.get(shape, f32, memory_space=Attribute.parse("2"))
379 print("memory space:", memref.memory_space)
390 print("memory space:", memref_layout.memory_space)
/llvm-project-15.0.7/mlir/test/Dialect/Bufferization/
H A Dinvalid.mlir85 …// expected-error @+1{{'bufferization.alloc_tensor' op attribute 'memory_space' failed to satisfy …
86 %0 = bufferization.alloc_tensor(%sz) {memory_space = "foo"} : tensor<?xf32>
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/
H A DBufferizationOps.td49 The optional `memory_space` attribute specifies the memory space when
51 If neither `copy` nor `memory_space` is specified, the default memory space
73 OptionalAttr<UI64Attr>:$memory_space);
134 // Build an op without `copy` or `memory_space`.
137 // Build an op without `memory_space`.
/llvm-project-15.0.7/mlir/python/mlir/_mlir_libs/_mlir/
H A Dir.pyi797 def memory_space(self) -> Attribute: ...
1054 def memory_space(self) -> Attribute: ...