Desarrollo de operadores de IA personalizados con CANN TIK y TVM

Introducción al desarrollo de operadores personalizados

• ¿Por qué desarrollar operadores personalizados? Casos de uso y restricciones
• Estructura del tiempo de ejecución de CANN y puntos de integración de operadores
• Visión general de TBE, TIK y TVM en el ecosistema de IA de Huawei

Uso de TIK para la programación de operadores de bajo nivel

• Comprensión del modelo de programación de TIK y las APIs soportadas
• Gestión de memoria y estrategia de tiling en TIK
• Creación, compilación y registro de un operador personalizado con CANN

Pruebas y validación de operadores personalizados

• Pruebas unitarias y de integración de operadores en el grafo
• Depuración de problemas de rendimiento a nivel de kernel
• Visualización de la ejecución del operador y el comportamiento de los buffers

Programación y optimización basada en TVM

• Visión general de TVM como compilador para operadores de tensors
• Escritura de una programación (schedule) para un operador personalizado en TVM
• Ajuste, pruebas de rendimiento y generación de código de TVM para Ascend

Integración con frameworks y modelos

• Registro de operadores personalizados para MindSpore y ONNX
• Verificación de la integridad del modelo y el comportamiento de respaldo (fallback)
• Soporte para grafos de múltiples operadores con precisión mixta

Estudios de caso y optimizaciones especializadas

• Estudio de caso: convolución de alta eficiencia para formas de entrada pequeñas
• Estudio de caso: optimización de operadores de atención con conciencia de memoria
• Mejores prácticas en el despliegue de operadores personalizados entre dispositivos

Resumen y próximos pasos