SCsub

#!/usr/bin/env python

import os
Import('env')
Import('env_modules')
cwd = os.getcwd()
env_rwkv = env_modules.Clone()
# get cwd
cenv = os.getcwd()

env_rwkv.add_source_files(env.modules_sources, "./rwkv.cuh/src/cpuops.cpp")

env_rwkv.add_source_files(env.modules_sources, "*.cpp")

if(env["PLATFORM"] != "win32"):
    

    # add rwkv.cuh/src/cpuops.cpp

    
    
    env_rwkv.Append(CCFLAGS=['-fexceptions'])
    
    # add -march=avx512
    print(env["platform"])
    if env["platform"] == "linuxbsd":
        print("Using linuxbsd")
        # add -O3
        env_rwkv.Append(CCFLAGS=['-O3'])
        # add -march=avx512
        env_rwkv.Append(CCFLAGS=['-march=core-avx2'])
        
        # add fast math
        env_rwkv.Append(CCFLAGS=['-ffast-math'])
   
    else:
        # arm
        print(env["platform"])
        print(env["arch"])
        if env["arch"] == "arm32":
            print("arm32")
            
            
        else:
            print("arm64")
            env_rwkv.Append(CCFLAGS=['-march=armv8.2-a+bf16'])
            # if platform is macos, set it to ignore all warnaserror
            if env["platform"] == "macos":
                env_rwkv.Append(CCFLAGS=['-Wno-error'])
    
    
    
    
    
    
    # add ./rwkv.hpp/include to include path
    # get current file location
    __file__ = env_rwkv.File('./SCsub').abspath
    path = os.path.dirname(os.path.abspath(__file__))
    env_rwkv.Append(CPPPATH=[path + "/rwkv.cuh/include"])

else:
    env_rwkv.Append(CCFLAGS=['/O2'])
    env_rwkv.Append(CCFLAGS=['/EHsc'])
    
    # add ./rwkv.cuh/include to include path
    # get current file location in scope of scons
    __file__ = env_rwkv.File('./SCsub').abspath
    path = os.path.dirname(os.path.abspath(__file__))
    env_rwkv.Append(CPPPATH=[path + "/rwkv.cuh/include"])
    
    # set avx512 
    env_rwkv.Append(CCFLAGS=['/arch:AVX2'])
        

    env_rwkv.add_source_files(env.modules_sources, "*.cpp")