Skip to main content
Ctrl+K

tvm-tirx

Get Started

  • Quick Start

IR Guides

  • IR Overview
  • Types and Expressions
  • Statements
  • Operators, Calls, and Regions
  • Functions and Modules
  • Printing and Parsing
  • Structural Traversal, Rewriting, and Comparison
  • Symbolic Integer Analysis
  • Working with IR in C++
  • Working with IR in Rust

Programming Guides

  • Programming Overview
  • Language Basics
  • Control Flow
  • Types, Tensors, and Memory
  • Metaprogramming
  • Calling CUDA and PTX
  • Data Layout
  • Tile Operations
  • Compiling and Inspecting
  • Profiling with IKET

Extending TIRx: everything is a plugin

  • Getting Started with Rust Extensions
  • Defining Types, Operators, and Passes
  • Using and Compiling the Extension

Reference

  • Python API
  • C++ API
  • Rust API

Developer Manual

  • Build from Source
  • Build This Doc Site
  • CUDA Backend Development
  • Reproduce CI/CD
  • Release Process
  • .rst

File cooperative_tensor.h

Contents

  • Definition (tvm/backend/metal/op/cooperative_tensor.h)
  • Includes
  • Namespaces
  • Functions

File cooperative_tensor.h#

↰ Parent directory (tvm/backend/metal/op)

Metal-owned matrix operations.

Contents

  • Definition (tvm/backend/metal/op/cooperative_tensor.h)

  • Includes

  • Namespaces

  • Functions

Definition (tvm/backend/metal/op/cooperative_tensor.h)#

  • Program Listing for File cooperative_tensor.h

Includes#

  • tvm/ir/op.h (File op.h)

  • tvm/ir/prim/expr.h (File expr.h)

Namespaces#

  • Namespace tvm

  • Namespace tvm::backend

  • Namespace tvm::backend::metal

Functions#

  • Function tvm::backend::metal::cooperative_tensor_fill_op

  • Function tvm::backend::metal::cooperative_tensor_load_op

  • Function tvm::backend::metal::cooperative_tensor_multiply_accumulate_op

  • Function tvm::backend::metal::cooperative_tensor_store_op

Contents
  • Definition (tvm/backend/metal/op/cooperative_tensor.h)
  • Includes
  • Namespaces
  • Functions

By Apache TVM Community