CUTLASS 2.1 (#83)
CUTLASS 2.1 contributes: - BLAS-style host-side API added to CUTLASS Library - Planar Complex GEMM kernels targeting Volta and Turing Tensor Cores - Minor enhancements and bug fixes
This commit is contained in:
@@ -105,11 +105,15 @@ public:
|
||||
/// allocating tensors.
|
||||
bool enabled;
|
||||
|
||||
/// If true, data distribution is set by the user and is not allowed to change
|
||||
/// If false, data distribution is allowed to change based on element_type (library::NumericTypeID)
|
||||
bool fix_data_distribution;
|
||||
|
||||
/// Data distribution for input tensors
|
||||
Distribution data_distribution;
|
||||
|
||||
/// Source of random tensor elements
|
||||
Provider provider;
|
||||
library::Provider provider;
|
||||
|
||||
/// Random number generator seed.
|
||||
int seed;
|
||||
@@ -162,10 +166,10 @@ public:
|
||||
void print_options(std::ostream &out, int indent = 0) const;
|
||||
|
||||
/// Returns true if a provider is enabled
|
||||
bool provider_enabled(Provider provider) const;
|
||||
bool provider_enabled(library::Provider provider) const;
|
||||
|
||||
/// Returns the index of a provider if its enabled
|
||||
size_t index(Provider provider) const;
|
||||
size_t index(library::Provider provider) const;
|
||||
};
|
||||
|
||||
/// Options related to profiling
|
||||
@@ -196,10 +200,10 @@ public:
|
||||
void print_options(std::ostream &out, int indent = 0) const;
|
||||
|
||||
/// Returns true if a provider is enabled
|
||||
bool provider_enabled(Provider provider) const;
|
||||
bool provider_enabled(library::Provider provider) const;
|
||||
|
||||
/// Returns the index of a provider if its enabled
|
||||
size_t index(Provider provider) const;
|
||||
size_t index(library::Provider provider) const;
|
||||
};
|
||||
|
||||
/// Options related to reporting
|
||||
|
||||
Reference in New Issue
Block a user