Desarrollo de operadores de IA personalizados con CANN TIK y TVM

Introducción al desarrollo de operadores personalizados

• ¿Por qué crear operadores personalizados? Casos de uso y restricciones
• Estructura del tiempo de ejecución (runtime) 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 a bajo nivel

• Comprensión del modelo de programación de TIK y las API compatibles
• Gestión de memoria y estrategia de bloques (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 núcleo (kernel)
• Visualización de la ejecución de operadores y el comportamiento de los búferes

Programación (scheduling) y optimización basadas en TVM

• Visión general de TVM como compilador para operaciones de tensores
• Escritura de un programa de scheduling (schedule) para un operador personalizado en TVM
• Ajuste (tuning), benchmarking y generación de código de TVM para Ascend

Integración con marcos de trabajo (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 con múltiples operadores en 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 operaciones personalizadas en distintos dispositivos

Resumen y próximos pasos