Get in Touch
 Duration 14 hours

Course Outline

Foundations of Custom Operator Development

  • Rationale for building custom operators, including specific use cases and constraints.
  • Analysis of CANN runtime structure and key operator integration points.
  • Overview of TBE, TIK, and TVM within the broader Huawei AI ecosystem.

Low-Level Operator Programming with TIK

  • Exploring the TIK programming model and its supported APIs.
  • Managing memory and implementing tiling strategies within TIK.
  • Processes for creating, compiling, and registering custom operators with CANN.

Validation and Testing of Custom Operators

  • Conducting unit and integration tests for operators within the execution graph.
  • Troubleshooting kernel-level performance bottlenecks.
  • Visualizing operator execution flows and buffer behaviors.

Scheduling and Optimization via TVM

  • Understanding TVM’s role as a compiler for tensor operations.
  • Crafting custom operator schedules within the TVM framework.
  • Applying TVM for tuning, benchmarking, and code generation specific to Ascend.

Framework and Model Integration

  • Registering custom operators for compatibility with MindSpore and ONNX.
  • Ensuring model integrity and managing fallback behaviors effectively.
  • Managing multi-operator graphs with mixed precision support.

Case Studies and Advanced Optimization Techniques

  • Case study: Optimizing high-efficiency convolution for small input shapes.
  • Case study: Enhancing memory-aware attention operator performance.
  • Best practices for deploying custom operators across various devices.

Conclusions and Future Directions

Requirements

  • Profound understanding of AI model architectures and operator-level computations.
  • Hands-on experience with Python and Linux development environments.
  • Familiarity with neural network compilers or graph-level optimization techniques.

Target Audience

  • Compiler engineers specializing in AI toolchain development.
  • Systems developers dedicated to low-level AI performance tuning.
  • Developers creating custom operators or targeting emerging AI workloads.

Custom Corporate Training

Training solutions designed exclusively for businesses.

  • Customized Content: We adapt the syllabus and practical exercises to the real goals and needs of your project.
  • Flexible Schedule: Dates and times adapted to your team's agenda.
  • Format: Online (live), In-company (at your offices), or Hybrid.
Investment

Price per private group, online live training, starting from 3200 € + VAT*

Contact us for an exact quote and to hear our latest promotions

Provisional Upcoming Courses (Contact Us For More Information)

Related Categories