dispatch_cutlass.h

#pragma once

#include "common.h"
#include "Tensor.h"

#include <cutlass/cutlass.h>
#include <cutlass/half.h>
#include <cutlass/bfloat16.h>

template<typename F>
inline void dispatchF16(Tensor::ScalarType type, F &&func) {
    if (type == Tensor::FP16) {
        func.template operator()<cutlass::half_t>();
    } else if (type == Tensor::BF16) {
        func.template operator()<cutlass::bfloat16_t>();
    } else {
        assert(false);
    }
}