Skip to content

Instantly share code, notes, and snippets.

@aleozlx

aleozlx/__blog_learning_cutlass_dtype.cpp

Created August 4, 2021 02:49
Show Gist options
  • Save aleozlx/4b36e752a4b1c5bc84b99a27c9fa6409 to your computer and use it in GitHub Desktop.
Save aleozlx/4b36e752a4b1c5bc84b99a27c9fa6409 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
__blog_learning_cutlass_dtype.cpp
using ElementAccumulator = float;
using ElementInputA = cutlass::half_t;
using ElementInputB = cutlass::half_t;
using ElementOutput = float;
using LayoutInputA = cutlass::layout::TensorNHWC;
using LayoutInputB = cutlass::layout::TensorNHWC;
using LayoutOutput = cutlass::layout::TensorNHWC;
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment