[nnc] Enable fusion of bfloat16 ops (#64196)
authorBert Maher <bertrand@fb.com>
Tue, 31 Aug 2021 03:08:15 +0000 (20:08 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Tue, 31 Aug 2021 03:09:36 +0000 (20:09 -0700)
commite7fb35021aca4b29bbf6f7120c5bff1400179175
tree8266878de5c66e85e142f1c88e0cbbad31c9b218
parent538647fe1fb94b7822ea3b8bbbd6901961431d60
[nnc] Enable fusion of bfloat16 ops (#64196)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64196

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D30643864

Pulled By: bertmaher

fbshipit-source-id: e95edeaf7089464d713ea1d1f951743d3e5f61c5
31 files changed:
test/test_jit_fuser_te.py
torch/csrc/jit/passes/tensorexpr_fuser.cpp
torch/csrc/jit/tensorexpr/block_codegen.cpp
torch/csrc/jit/tensorexpr/codegen.cpp
torch/csrc/jit/tensorexpr/codegen.h
torch/csrc/jit/tensorexpr/cpp_codegen.cpp
torch/csrc/jit/tensorexpr/cuda_codegen.cpp
torch/csrc/jit/tensorexpr/eval.cpp
torch/csrc/jit/tensorexpr/eval.h
torch/csrc/jit/tensorexpr/expr.cpp
torch/csrc/jit/tensorexpr/expr.h
torch/csrc/jit/tensorexpr/fwd_decls.h
torch/csrc/jit/tensorexpr/half_support.h
torch/csrc/jit/tensorexpr/hash_provider.h
torch/csrc/jit/tensorexpr/ir.cpp
torch/csrc/jit/tensorexpr/ir.h
torch/csrc/jit/tensorexpr/ir_cloner.cpp
torch/csrc/jit/tensorexpr/ir_cloner.h
torch/csrc/jit/tensorexpr/ir_mutator.cpp
torch/csrc/jit/tensorexpr/ir_mutator.h
torch/csrc/jit/tensorexpr/ir_printer.cpp
torch/csrc/jit/tensorexpr/ir_printer.h
torch/csrc/jit/tensorexpr/ir_simplifier.h
torch/csrc/jit/tensorexpr/ir_visitor.cpp
torch/csrc/jit/tensorexpr/ir_visitor.h
torch/csrc/jit/tensorexpr/kernel.cpp
torch/csrc/jit/tensorexpr/llvm_codegen.cpp
torch/csrc/jit/tensorexpr/reduction.h
torch/csrc/jit/tensorexpr/tensorexpr_init.cpp
torch/csrc/jit/tensorexpr/types.cpp
torch/csrc/jit/tensorexpr/types.h