mirror of https://github.com/llvm/torch-mlir
25 lines
617 B
Python
25 lines
617 B
Python
|
# -*- Python -*-
|
||
|
|
# This file is licensed under a pytorch-style license
|
||
|
|
# See frontends/pytorch/LICENSE for license information.
|
||
|
|
# RUN: python %s | FileCheck %s
|
||
|
|
|
||
|
|
import torch
|
||
|
|
import _torch_mlir as m
|
||
|
|
|
||
|
|
t0 = torch.randn((4,4))
|
||
|
|
t1 = torch.randn((4,4))
|
||
|
|
|
||
|
|
with m.c10.AcapController() as c:
|
||
|
|
result = t0 + t1
|
||
|
|
|
||
|
|
result = result * t0
|
||
|
|
|
||
|
|
# NOTE: Ops involved with printing throw RuntimeError about calling a kernel
|
||
|
|
# from an unboxed API.
|
||
|
|
print(result)
|
||
|
|
|
||
|
|
# CHECK: CAPTURE: aten::add.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> (Tensor)
|
||
|
|
# CHECK-NOT: CAPTURE: aten::mul
|
||
|
|
log = c.get_debug_log()
|
||
|
|
for line in log: print(line)
|