llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make TMTensor tests pass.

pt2_phase1
Stella Laurenzo 2023-11-01 18:49:36 -07:00
parent af20d259b9
commit 6ab6d01da1
9 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CMakeLists.txt
View File

@ -164,7 +164,6 @@ add_subdirectory(tools)
add_custom_target(check-torch-mlir-all) add_custom_target(check-torch-mlir-all)
add_dependencies(check-torch-mlir-all add_dependencies(check-torch-mlir-all
check-torch-mlir check-torch-mlir
check-torch-mlir-dialects
check-torch-mlir-capi check-torch-mlir-capi
) )

1
include/CMakeLists.txt
View File

@ -1 +1,2 @@
add_subdirectory(torch-mlir) add_subdirectory(torch-mlir)
add_subdirectory(torch-mlir-dialects)

1
lib/Dialect/CMakeLists.txt
View File

@ -1,2 +1,3 @@
add_subdirectory(TMTensor)
add_subdirectory(Torch) add_subdirectory(Torch)
add_subdirectory(TorchConversion) add_subdirectory(TorchConversion)

2
lib/Dialect/TMTensor/IR/CMakeLists.txt
View File

@ -26,4 +26,4 @@ add_mlir_library(TorchMLIRTMTensorDialect
MLIRViewLikeInterface MLIRViewLikeInterface
) )
torch_mlir_dialects_target_includes(TorchMLIRTMTensorDialect) torch_mlir_target_includes(TorchMLIRTMTensorDialect)

2
lib/Dialect/TMTensor/Transforms/CMakeLists.txt
View File

@ -21,3 +21,5 @@ add_mlir_library(TorchMLIRTMTensorPasses
MLIRTensorDialect MLIRTensorDialect
MLIRTransforms MLIRTransforms
) )
torch_mlir_target_includes(TorchMLIRTMTensorPasses)

2
test/tmtensor/bufferize.mlir → test/Dialect/TMTensor/bufferize.mlir
View File

@ -1,4 +1,4 @@
// RUN: torch-mlir-dialects-opt -split-input-file -tm-tensor-bufferize %s | FileCheck %s // RUN: torch-mlir-opt -split-input-file -tm-tensor-bufferize %s | FileCheck %s
// ----- // -----
// CHECK-LABEL: func.func @scan_1d_inclusive( // CHECK-LABEL: func.func @scan_1d_inclusive(

2
test/tmtensor/canonicalize.mlir → test/Dialect/TMTensor/canonicalize.mlir
View File

@ -1,4 +1,4 @@
// RUN: torch-mlir-dialects-opt -canonicalize -split-input-file %s | FileCheck %s // RUN: torch-mlir-opt -canonicalize -split-input-file %s | FileCheck %s
// CHECK-LABEL: func.func @tensor.cast( // CHECK-LABEL: func.func @tensor.cast(
func.func @tensor.cast(%arg0: tensor<128xi32>) -> tensor<128xi32> { func.func @tensor.cast(%arg0: tensor<128xi32>) -> tensor<128xi32> {

2
test/tmtensor/convert_to_loops.mlir → test/Dialect/TMTensor/convert_to_loops.mlir
View File

@ -1,4 +1,4 @@
// RUN: torch-mlir-dialects-opt -split-input-file -tm-tensor-to-loops %s | FileCheck %s // RUN: torch-mlir-opt -split-input-file -tm-tensor-to-loops %s | FileCheck %s
func.func @scan_1d_inclusive(%0: memref<128xi32>, %1: memref<128xi32>) { func.func @scan_1d_inclusive(%0: memref<128xi32>, %1: memref<128xi32>) {
%c0 = memref.alloc() : memref<i32> %c0 = memref.alloc() : memref<i32>

2
test/tmtensor/invalid.mlir → test/Dialect/TMTensor/invalid.mlir
View File

@ -1,4 +1,4 @@
// RUN: torch-mlir-dialects-opt -split-input-file -verify-diagnostics %s // RUN: torch-mlir-opt -split-input-file -verify-diagnostics %s
func.func @scatter_mixed_tensor_memref( func.func @scatter_mixed_tensor_memref(
%update : memref<?x?xf32>, %indices : tensor<?x1xi32>, %update : memref<?x?xf32>, %indices : tensor<?x1xi32>,