aboutsummaryrefslogtreecommitdiff
path: root/cutlass-example/cutlass/gemm/gemm_shared_stream.h
diff options
context:
space:
mode:
Diffstat (limited to 'cutlass-example/cutlass/gemm/gemm_shared_stream.h')
-rw-r--r--cutlass-example/cutlass/gemm/gemm_shared_stream.h113
1 files changed, 0 insertions, 113 deletions
diff --git a/cutlass-example/cutlass/gemm/gemm_shared_stream.h b/cutlass-example/cutlass/gemm/gemm_shared_stream.h
deleted file mode 100644
index c6ff7bd..0000000
--- a/cutlass-example/cutlass/gemm/gemm_shared_stream.h
+++ /dev/null
@@ -1,113 +0,0 @@
-/***************************************************************************************************
- * Copyright (c) 2017-2018, NVIDIA CORPORATION. All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without modification, are permitted
- * provided that the following conditions are met:
- * * Redistributions of source code must retain the above copyright notice, this list of
- * conditions and the following disclaimer.
- * * Redistributions in binary form must reproduce the above copyright notice, this list of
- * conditions and the following disclaimer in the documentation and/or other materials
- * provided with the distribution.
- * * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used
- * to endorse or promote products derived from this software without specific prior written
- * permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
- * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
- * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
- * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
- * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
- * STRICT LIABILITY, OR TOR (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- **************************************************************************************************/
-/*! \file
- \brief Defines abstractions for managing loading and storing fragments to shared memory in the
- efficient GEMM pipeline.
-*/
-#pragma once
-
-#include <cutlass/gemm/gemm_shared_tile.h>
-
-namespace cutlass {
-namespace gemm {
-
-////////////////////////////////////////////////////////////////////////////////////////////////////
-
-template <
- /// The load iterator.
- typename Iterator_,
- /// The transformer to be applied after the data has been copied from shared memory.
- typename Transformer_ = Copy<typename Iterator_::Fragment> >
-
-struct SharedLoadStream {
- /// The load iterator.
- typedef Iterator_ Iterator;
- /// The transformer.
- typedef Transformer_ Transformer;
-
- /// The fragment that is copied from shared memory.
- typedef typename Iterator::Fragment FetchedFragment;
- /// The fragment that is obtained after the transformation by the transformer.
- typedef typename Transformer::OutputFragment TransformedFragment;
- /// Make sure the fragments match.
- static_assert((platform::is_same<FetchedFragment, typename Transformer::InputFragment>::value),
- "");
- /// The output fragment.
- typedef TransformedFragment Fragment;
-
- /// The params.
- struct Params {
- /// The iterator params.
- typename Iterator::Params iterator;
-
- /// Setup the params.
- CUTLASS_HOST_DEVICE int initialize() { return iterator.initialize(); }
- };
-
- /// The storage in shared memory needed by that stream.
- typedef typename Iterator::Storage SharedStorage;
-
- /// Ctor.
- CUTLASS_DEVICE SharedLoadStream() {}
-
- /// Ctor.
- CUTLASS_DEVICE SharedLoadStream(Params const &params, SharedStorage &shared_storage) {
- this->initialize(params, shared_storage);
- }
-
- /// Initialize the stream.
- CUTLASS_DEVICE void initialize(Params const &params, SharedStorage &shared_storage) {
- // The iterator.
- iterator = Iterator(params.iterator, shared_storage);
- // The transformer.
- transformer = Transformer();
- }
-
- /// Load the data from shared memory to the fetch fragment.
- CUTLASS_DEVICE void copy(FetchedFragment &fetched) { shared_iterator_load(iterator, fetched); }
-
- /// Load the data from shared memory to the fetch fragment.
- CUTLASS_DEVICE void copy(int d, FetchedFragment &fetched) {
- shared_iterator_load(iterator, fetched, d);
- }
-
- /// Commit the data.
- CUTLASS_DEVICE void commit(FetchedFragment &fetched, TransformedFragment &transformed) {
- transformer.transform(fetched, transformed);
- }
-
- /// Increment the stage.
- CUTLASS_DEVICE void inc_stage() { iterator.inc_stage(); }
-
- /// The iterator.
- Iterator iterator;
- /// The transformer.
- Transformer transformer;
-};
-
-////////////////////////////////////////////////////////////////////////////////////////////////////
-
-} // namespace gemm
-} // namespace cutlass