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

View File

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

View File

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

View File

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

View File

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

View File

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

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(

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(
func.func @tensor.cast(%arg0: tensor<128xi32>) -> tensor<128xi32> {

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>) {
%c0 = memref.alloc() : memref<i32>

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(
%update : memref<?x?xf32>, %indices : tensor<?x1xi32>,