File indexing completed on 2026-04-09 07:49:12
0001 #pragma once
0002
0003
0004
0005
0006
0007
0008
0009 enum {
0010 T_UNDEF,
0011 T_DISC
0012 };
0013
0014
0015 #if defined(__CUDACC__) || defined(__CUDABE__)
0016 #else
0017 #include <cstring>
0018
0019 struct torchtype
0020 {
0021 static constexpr const char* T_UNDEF_ = "undef";
0022 static constexpr const char* T_DISC_ = "disc" ;
0023
0024 static unsigned Type(const char* name);
0025 static const char* Name(unsigned type);
0026 };
0027
0028 inline unsigned torchtype::Type(const char* name)
0029 {
0030 unsigned type = T_UNDEF ;
0031 if(strcmp(name,T_DISC_)==0) type = T_DISC ;
0032 return type ;
0033 }
0034
0035 inline const char* torchtype::Name(unsigned type)
0036 {
0037 const char* n = T_UNDEF_ ;
0038 switch(type)
0039 {
0040 case T_UNDEF: n = T_UNDEF_ ; break ;
0041 case T_DISC: n = T_DISC_ ; break ;
0042 }
0043 return n ;
0044 }
0045 #endif
0046
0047