Home / Class/ bool Class — pytorch Architecture

bool Class — pytorch Architecture

Architecture documentation for the bool class in ComputeSparseTile.h from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/native/sparse/cuda/ComputeSparseTile.h lines 16–31

template <typename Element, typename Pointwise>
struct TileValueOrderedT {
  union {
    struct {
      Element value;
      uint2b_t col;
      uint2b_t row;
    } parts;
    uint32_t raw;
  };
  CUTLASS_DEVICE bool operator<(
      TileValueOrderedT<Element, Pointwise> const& other) const {
    return Pointwise::apply(parts.value) < Pointwise::apply(other.parts.value);
  }
  CUTLASS_DEVICE TileValueOrderedT() {}
};

Analyze Your Own Codebase

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

Try Supermodel Free