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

[cleanup] Be consistent about apostrophe

pull/1571/head
Sean Silva 2022-11-10 10:26:22 +00:00
parent a7ac0def45
commit cc468d2d16
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/test/compile_api/tracing.py
View File

@ -59,14 +59,14 @@ class DictModule(torch.nn.Module):
try: try:
# CHECK: Only Tensors, TensorPlaceholder's, or sequences of Tensors and TensorPlaceholder's are supported as example args for method inputs. Got '{'a': tensor(3.)}' # CHECK: Only Tensor's, TensorPlaceholder's, or sequences of Tensor's and TensorPlaceholder's are supported as example args for method inputs. Got '{'a': tensor(3.)}'
torch_mlir.compile(DictModule(), {'a': torch.tensor(3.0)}, use_tracing=True) torch_mlir.compile(DictModule(), {'a': torch.tensor(3.0)}, use_tracing=True)
except Exception as e: except Exception as e:
print(e) print(e)
try: try:
# CHECK: Only Tensors, TensorPlaceholder's, or sequences of Tensors and TensorPlaceholder's are supported as example args for method inputs. Got '{'a': tensor(3.)}' # CHECK: Only Tensor's, TensorPlaceholder's, or sequences of Tensor's and TensorPlaceholder's are supported as example args for method inputs. Got '{'a': tensor(3.)}'
torch_mlir.compile(DictModule(), [{'a': torch.tensor(3.0)}], use_tracing=True) torch_mlir.compile(DictModule(), [{'a': torch.tensor(3.0)}], use_tracing=True)
except Exception as e: except Exception as e:
print(e) print(e)

4
python/torch_mlir/__init__.py
View File

@ -173,8 +173,8 @@ class ExampleArgs:
example_args = [example_args] example_args = [example_args]
for arg in example_args: for arg in example_args:
if not isinstance(arg, _example_arg): if not isinstance(arg, _example_arg):
raise Exception(f"Only Tensors, TensorPlaceholder's, or sequences of " raise Exception(f"Only Tensor's, TensorPlaceholder's, or sequences of "
f"Tensors and TensorPlaceholder's are supported as " f"Tensor's and TensorPlaceholder's are supported as "
f"example args for method inputs. " f"example args for method inputs. "
f"Got '{arg}'.") f"Got '{arg}'.")
return tuple(example_args) return tuple(example_args)