Can I use float atomics on gpu?
Asked on 2024-07-30
1 search
Yes, you can use float atomics on the GPU. This feature is part of the updates presented in the session Train your machine learning and AI models on Apple GPUs. The session covers various improvements and features for training machine learning models on Apple GPUs, including support for mixed precision and new data types like Pfloat 16, which are beneficial for tasks such as mixed precision training.
For more details, you can refer to the chapter on JAX features in the session Train your machine learning and AI models on Apple GPUs.
Train your machine learning and AI models on Apple GPUs
Learn how to train your models on Apple Silicon with Metal for PyTorch, JAX and TensorFlow. Take advantage of new attention operations and quantization support for improved transformer model performance on your devices.
Build a spatial drawing app with RealityKit
Harness the power of RealityKit through the process of building a spatial drawing app. As you create an eye-catching spatial experience that integrates RealityKit with ARKit and SwiftUI, you’ll explore how resources work in RealityKit and how to use features like low-level mesh and texture APIs to achieve fast updates of the users’ brush strokes.
Accelerate machine learning with Metal
Learn how to accelerate your machine learning transformer models with new features in Metal Performance Shaders Graph. We’ll also cover how to improve your model’s compute bandwidth and quality, and visualize it in the all new MPSGraph viewer.