drbh

feat: build flash mla with kernel builder

1f83cde over 1 year ago

316 Bytes

#pragma once

#include "cutlass/barrier.h"

namespace flash {

////////////////////////////////////////////////////////////////////////////////////////////////////
// Enumerates the reserved named barriers to avoid potential conflicts

enum class NamedBarriers {
    SReady = 1,
    SoftmaxReady = 2,
};

} // flash

	#pragma once

	#include "cutlass/barrier.h"

	namespace flash {

	////////////////////////////////////////////////////////////////////////////////////////////////////
	// Enumerates the reserved named barriers to avoid potential conflicts

	enum class NamedBarriers {
	SReady = 1,
	SoftmaxReady = 2,
	};

	} // flash