xref: /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/intrinsic_symbols.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 
3 #ifdef TORCH_ENABLE_LLVM
4 #include <c10/util/ArrayRef.h>
5 
6 namespace torch {
7 namespace jit {
8 namespace tensorexpr {
9 
10 struct SymbolAddress {
11   const char* symbol;
12   void* address;
13 
SymbolAddressSymbolAddress14   SymbolAddress(const char* sym, void* addr) : symbol(sym), address(addr) {}
15 };
16 
17 c10::ArrayRef<SymbolAddress> getIntrinsicSymbols();
18 
19 } // namespace tensorexpr
20 } // namespace jit
21 } // namespace torch
22 #endif // TORCH_ENABLE_LLVM
23