pytorch Architecture Docs
Home / Class/ type Class — pytorch Architecture

type Class — pytorch Architecture

Architecture documentation for the type class in atomic.h from the pytorch codebase.

Class c

Entity Profile

Source Code

c10/metal/atomic.h lines 12–19

template <>
struct AtomicType<float> {
  using type = ::metal::atomic<float>;
  static inline void atomic_add(device type* data, long offset, float value) {
    ::metal::atomic_fetch_add_explicit(
        data + offset, value, ::metal::memory_order_relaxed);
  }
};

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free