| /llvm-project-15.0.7/mlir/include/mlir/Dialect/AMX/ |
| H A D | CMakeLists.txt | 1 add_mlir_dialect(AMX amx) 2 add_mlir_doc(AMX AMX Dialects/ -gen-dialect-doc -dialect=amx) 4 set(LLVM_TARGET_DEFINITIONS AMX.td)
|
| H A D | AMX.td | 1 //===-- AMX.td - AMX dialect operation definitions *- tablegen -*----------===// 9 // This file defines the basic operations for the AMX dialect. 11 // The Intel Advanced Matrix Extensions (AMX) provide a tile matrix 28 #ifndef AMX 29 #define AMX 35 // AMX dialect definition. 48 The dialect is split into user-facing AMX ops (AMX_Op) and 63 // AMX Op and IntrOp definitions. 76 // AMX Op definitions (user facing). 253 // AMX IntrOp definitions (LLVM compiler facing). [all …]
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/Vector/CPU/AMX/ |
| H A D | lit.local.cfg | 3 # AMX tests must be enabled via build flag. 12 # Run test in emulator (Intel SDE): AMX needs Sapphire Rapids CPU.
|
| H A D | test-mulf-full.mlir | 9 // Note: To run this test, your CPU must support AMX.
|
| H A D | test-muli-full.mlir | 9 // Note: To run this test, your CPU must support AMX.
|
| H A D | test-tilezero-block.mlir | 6 // Note: To run this test, your CPU must support AMX.
|
| H A D | test-tilezero.mlir | 6 // Note: To run this test, your CPU must support AMX.
|
| H A D | test-mulf.mlir | 6 // Note: To run this test, your CPU must support AMX.
|
| H A D | test-muli.mlir | 6 // Note: To run this test, your CPU must support AMX.
|
| H A D | test-muli-ext.mlir | 6 // Note: To run this test, your CPU must support AMX.
|
| /llvm-project-15.0.7/llvm/test/Transforms/InstCombine/X86/ |
| H A D | x86-amx-load-store.ll | 24 ; CHECK-NEXT: [[AMX:%.*]] = tail call x86_amx @llvm.x86.tileloadd64.internal(i16 16, i16 16, i8*… 25 ; CHECK-NEXT: [[BC:%.*]] = bitcast x86_amx [[AMX]] to <256 x i32>
|
| /llvm-project-15.0.7/mlir/lib/Dialect/AMX/IR/ |
| H A D | CMakeLists.txt | 5 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/AMX
|
| /llvm-project-15.0.7/mlir/lib/Target/LLVMIR/Dialect/ |
| H A D | CMakeLists.txt | 3 add_subdirectory(AMX)
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/ |
| H A D | CMakeLists.txt | 7 add_subdirectory(AMX)
|
| /llvm-project-15.0.7/mlir/lib/Dialect/ |
| H A D | CMakeLists.txt | 7 add_subdirectory(AMX)
|
| /llvm-project-15.0.7/llvm/test/CodeGen/X86/ |
| H A D | O0-pipeline.ll | 21 ; CHECK-NEXT: Lower AMX intrinsics 22 ; CHECK-NEXT: Lower AMX type for load/store
|
| H A D | opt-pipeline.ll | 30 ; CHECK-NEXT: Lower AMX intrinsics 31 ; CHECK-NEXT: Lower AMX type for load/store
|
| /llvm-project-15.0.7/llvm/lib/Target/X86/ |
| H A D | X86InstrAMX.td | 1 //===---- X86InstrAMX.td - AMX Instruction Set Extension --*- tablegen -*--===// 9 // This file describes the instructions that make up the Intel AMX instruction 15 // AMX instructions
|
| /llvm-project-15.0.7/clang/include/clang/Basic/ |
| H A D | BuiltinsX86_64.def | 108 // AMX internal builtin 119 // AMX
|
| /llvm-project-15.0.7/llvm/include/llvm/IR/ |
| H A D | Intrinsics.h | 143 AMX, enumerator
|
| /llvm-project-15.0.7/llvm/test/CodeGen/X86/AMX/ |
| H A D | amx-fastpreconfig.mir | 4 # Test the case which has TILELOADD being mixed in psuedo AMX instruction
|
| /llvm-project-15.0.7/utils/bazel/llvm-project-overlay/mlir/ |
| H A D | BUILD.bazel | 1569 # AMX dialect. 1574 srcs = ["include/mlir/Dialect/AMX/AMX.td"], 1602 "include/mlir/Dialect/AMX/AMX.h.inc", 1606 "include/mlir/Dialect/AMX/AMX.cpp.inc", 1610 "g3doc/Dialects/AMX/AMX.md", 1614 td_file = "include/mlir/Dialect/AMX/AMX.td", 1620 srcs = ["lib/Dialect/AMX/IR/AMXDialect.cpp"], 1659 td_file = "include/mlir/Dialect/AMX/AMX.td", 3376 "lib/Dialect/LLVMIR/IR/*AMX*.cpp", 3377 "lib/Dialect/LLVMIR/IR/*AMX*.h", [all …]
|
| /llvm-project-15.0.7/mlir/test/ |
| H A D | CMakeLists.txt | 34 option(MLIR_RUN_AMX_TESTS "Run AMX tests.")
|
| /llvm-project-15.0.7/llvm/lib/IR/ |
| H A D | Function.cpp | 1016 OutputTable.push_back(IITDescriptor::get(IITDescriptor::AMX, 0)); in DecodeIITType() 1283 case IITDescriptor::AMX: return Type::getX86_AMXTy(Context); in DecodeFixedType() 1469 case IITDescriptor::AMX: return !Ty->isX86_AMXTy(); in matchIntrinsicType()
|
| /llvm-project-15.0.7/mlir/include/mlir/Conversion/ |
| H A D | Passes.td | 911 (AMX, X86Vector, ArmNeon, ArmSVE, etc.) in combination with the 928 "Enables the use of AMX dialect while lowering the vector "
|