Fixes for public issue #265

2021-05-19 10:16:52 -07:00
parent b68113f5be
commit da2f110906
5 changed files with 93 additions and 68 deletions
@@ -69,6 +69,7 @@ enum
 #include <cuda_fp16.h>

 #include "cutlass/cutlass.h"
+#include "cutlass/platform/platform.h"

 ///////////////////////////////////////////////////////////////////////////////////////////////////

@@ -545,9 +546,9 @@ half_t copysign(half_t const& a, half_t const& b) {
 //
 ///////////////////////////////////////////////////////////////////////////////////////////////////

-namespace std {
+namespace cutlass {
+namespace platform {    

-#if !defined(__CUDACC_RTC__)
 /// Numeric limits
 template <>
 struct numeric_limits<cutlass::half_t> {
@@ -593,9 +594,9 @@ struct numeric_limits<cutlass::half_t> {
  /// Returns smallest finite value
  static cutlass::half_t denorm_min() { return cutlass::half_t::bitcast(0x0001); }
 };
-#endif

-}  // namespace std
+}  // namespace platform
+}  // namespace cutlass

 ///////////////////////////////////////////////////////////////////////////////////////////////////
 //